File "E:\ZLF\edm-main\torch_utils\distributed.py", line 30, in init torch.cuda.set_device(int(os.environ.get('LOCAL_RANK', '0'))) File "D:\anoconda\envs\DG\lib\site-packages\torch\cuda\__init__.py", line 350, in set_device torch._C._cuda_setDevice(device) AttributeError: module 'torch._C' has no attribute '_cuda_setDevice'怎么改
时间: 2023-07-21 15:54:00 浏览: 319
ex_07b_ack_data_rx.rar_ex_zlf
这个错误似乎是由于 PyTorch 版本不兼容或安装方式不正确导致的。你可以尝试以下几种方法来解决这个问题:
1. 检查 PyTorch 版本是否正确,如果不正确,可以尝试升级到最新版本或更换兼容的版本。
2. 检查 PyTorch 是否正确安装,可以尝试重新安装 PyTorch。
3. 检查 CUDA 是否正确安装,可以尝试重新安装 CUDA。
4. 检查环境变量是否正确设置,可以尝试手动设置环境变量。
如果以上方法都不能解决问题,建议查看 PyTorch 的官方文档或向 PyTorch 社区寻求帮助。
阅读全文