问题:错误代码为"cudaErrorInvalidDevice",表示设备顺序无效。
原因:这可能是由于设备驱动程序或CUDA版本不兼容导致的。可以尝试更新CUDA驱动程序或安装支持的CUDA版本的PyTorch版本。
两张方案解决:
1.这是torch1.8.x的bug,需要将pytorch版本升为1.9.0就可以解决了
2.训练的时候指定显卡进行训练,在训练代码中添加代码
os.environ['CUDA_VISIBLE_DEVICES'] = '1'
问题:错误代码为"cudaErrorInvalidDevice",表示设备顺序无效。
原因:这可能是由于设备驱动程序或CUDA版本不兼容导致的。可以尝试更新CUDA驱动程序或安装支持的CUDA版本的PyTorch版本。
两张方案解决:
1.这是torch1.8.x的bug,需要将pytorch版本升为1.9.0就可以解决了
2.训练的时候指定显卡进行训练,在训练代码中添加代码
os.environ['CUDA_VISIBLE_DEVICES'] = '1'