运用pytorch进行模型训练报错ImportError: cannot import name 'validate_cuda_device'

运用pytorch进行模型训练报错ImportError: cannot import name 'validate_cuda_device'

代码中DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

import torch

环境不对啊，或者你的cuda和cudnn版本匹配不上。请说下你的环境，包括显卡型号，cuda和cudnn版本，torch，torchvision和torchaudio这几个的版本，以及torch是否为GPU版本

望采纳！点击该回答右侧的“采纳”按钮即可采纳！！！
我觉得由于你的pytorch版本过低或者你的pytorch库与你的pytorch版本不兼容导致的。

你可以尝试升级您的pytorch版本或者重新安装pytorch库来解决这个问题。

你也可以在执行import torch语句之前添加如下代码来检测pytorch版本和是否存在cuda设备：

import sys
print(sys.version)
print(torch.version)
print(torch.cuda.is_available())