做网站挂靠百度爱采购排名

---

前言

---

问题：由于全局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，即可解决办法。

---

总结