20250513_问题:由于全局Pytorch导致的错误
文章目录
- 前言
- 问题:由于`全局Pytorch`导致的错误
- 找到问题(1)
- 找到问题(2)
- 总结
前言
问题:由于全局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,即可解决办法。