module 'torch._C' has no attribute '_cuda_setDevice'
时间: 2023-10-16 07:08:54 浏览: 201
导入FashionMNIST数据集时报错module ‘torchvision.datasets’ has no attribute ‘FashionMNIS’
根据提供的引用内容,当出现 "module 'torch._C' has no attribute '_cuda_setDevice'" 这个错误时,可能是由于使用了过时的方法来设置GPU设备。你可以尝试使用新的方法来设置GPU设备。例如,在你的Python文件中,可以使用以下代码来设置GPU设备:
torch.cuda.set_device(-1)
如果你是通过命令行执行Python文件,可以在最后添加以下命令来设置GPU设备:
python train.py --你的GPU的设置对应形参 -1
请注意,你需要将上述代码中的-1更改为你要使用的GPU设备的编号。此外,还要确保你的GPU版本已经正确安装并可用。
阅读全文