windows10+ Anaconda pyhton3.7+cuda10.0 PyTorch安装
PyTorch安装
环境:windows10 Anaconda pyhton3.7 cuda10.0
只需要在Get Started 直接选择自己的系统就可以
当你的cuda为10.0的时候你需要在后面找到cuda10.0的版本进行安装
当出现from torch._C import * ImportError: DLL load failed: 找不到指定的模块” 错误原因是安装的PyTorch版本与你的cuda
的版本不相符。解决方法pip uninstall torch chon重新下载和cuda相符的版本torch
安装成功后测试:
pyorch测试
import torch
x = torch.rand(5, 3)
print(x)
PyTorch是否启用了GPU驱动程序和CUDA并可以访问它
import torch
torch.cuda.is_available()