服务器上安装pytorch-gpu

1.创建虚拟环境

conda create -n pytorch-gpu python=3.6

如果需要查看当前有哪些虚拟环境

conda info --env

2.进入虚拟环境

如果启用创建的环境，输入下面指令

source activate pytorch-gpu
# or
conda activate pytorch-gpu

 退出当前虚拟环境的话，输入指令

conda deactivate

3.安装pytorch

首先要查看服务器的cuda版本，使用如下命令

如上图，cuda版本是10.1

或者输入命令

cat /usr/local/cuda/version.txt 

 

在刚刚创建的环境中安装pytorch, 在pytorch 官网 找到对应的下载指令进行下载，选择与自己环境匹配的 ，如下图是我的配置

在已经激活的环境中输入上图红框中的命令：

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

默认pytorch官网为下载源，下载速度太慢，很容易报错，所以更改为清华大学镜像，
命令行输入下面的命令：

# 添加Anaconda的清华镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
# 设置搜索时显示通道地址
conda config --set show_channel_urls yes  

然后在输入：

conda install pytorch torchvision cudatoolkit=10.0

注意要把后面的-c pytorch去掉，不然还是使用的默认源下载。

然后等待下载安装即可。

4.测试

不报错，并返回True, 说明安装成功！！

参考博客

https://blog.csdn.net/amnesiagreen/article/details/106349598

https://blog.csdn.net/kan2281123066/article/details/93789303

 

本文地址：https://blog.csdn.net/sarracode/article/details/107312975