admin管理员组

文章数量:1558098

1.创建虚拟环境

conda create -n pytorch-gpu python=3.6

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

conda info --env

2.进入虚拟环境

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

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

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

conda deactivate

删除虚拟环境的话，输入指令

conda remove -n pytorch-gpu --all

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/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu/anaconda/cloud/pytorch/
# 设置搜索时显示通道地址
conda config --set show_channel_urls yes

然后在输入：

conda install pytorch torchvision cudatoolkit=10.0

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

然后等待下载安装即可。

4.测试

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

参考博客

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

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

 

本文标签： 器上PytorchGPU

版权声明：本文标题：服务器上安装pytorch-gpu 内容由热心网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：https://m.elefans.com/xitong/1727354946a1110096.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。