1、安装pytorch
1.1、换源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
1.2、如何安装pytorch
pip install torch==1.1.0
pip install torch==0.4.0
pip install torch==1.0.0
pip install torch==1.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版本号