pip install torch1.0.0
pip install torch1.5.0
pip install torch==1.4.0
1.3、查看pytoch是否支持CUDA
命令行输入python进入代码执行界面,输入下面的语句。
torch.cuda.is_available()
返回True说明支持
以下列出对应版本号:
torch 1.1.0 ->CUDA 9.2
torch 1.2.0 ->CUDA 10.0
torch 1.3.0 ->CUDA 10.1
torch 1.4.0 ->CUDA 10.2
torch 1.5.0 ->CUDA 10.2
2、安装CUDA
暂时不会,建议参考其他博客,安装完成,输入
nvcc -V
查看CUDA版本号