前言
问题:由于全局Pytorch导致的错误
代码报错显示无法使用CUDA
AttributeError: module ‘torch._C’ has no attribute ‘_cuda_setDevice’
找到问题(1)
- import torch
- print('是否可用CUDA',torch.cuda.is_available())
- print("PyTorch Version:", torch.__version__)
- # 如果 CUDA 可用,查看 CUDA 版本
- if torch.cuda.is_available():
- print("CUDA Version:", torch.version.cuda)
- print("当前 GPU 设备:", torch.cuda.get_device_name(0))
- else:
- print("CUDA 不可用!")
复制代码 正确的显示:
找到问题(2)
- python -c "import torch; print(torch.__version__); print(torch.__file__)"
复制代码 确保输出的路径是假造环境中的路径,而不是全局路径。例如,它应该类似于:
- D:\Anaconda\envs\myenv-cvrp\lib\site-packages\torch
复制代码 手动删除全局安装的残留文件,路径为:
- C:\Users\Admin\AppData\Roaming\Python\Python38\site-packages\torch
复制代码 注:只要手动删除全局pytorch,即可办理办法。
总结
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。
|