安装pytorch

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 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版本号