服务器上安装pytorch-gpu
- 作者: Black66506319
- 来源: 51数据库
- 2022-05-08
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 http://www.51sjk.com/Upload/Articles/1/0/310/310598_20210801001034214.jpg
conda config --add channels http://www.51sjk.com/Upload/Articles/1/0/310/310598_2021080100103421401.jpg
conda config --add channels http://www.51sjk.com/Upload/Articles/1/0/310/310598_20210801001034230.jpg
# 设置搜索时显示通道地址
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
?
推荐阅读
