在Linux服务器上安装cuda11版pytorch

0.预备工作:安装anaconda
清华镜像网站上找到自己想要的安装包,比如我下的是下图红框里的那个:

在Linux服务器上安装cuda11版pytorch
用右键复制链接地址,然后用wget -P指定安装包的安装路径即可:
wget -P target_path https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-4.3.1-Linux-x86_64.sh

接着进入这个target_path中,用 bash Anaconda3-4.3.1-Linux-x86_64.sh安装,安装完成后还有两个步骤需要完成:
1)输入 vi ~/.bashrc,在文件的最后添加 export PATH=~/anaconda3/bin:$PATH,接着用wq保存修改并退出
2)用 source ~/.bashrc刷新一下环境变量即可。

1.创建一个新的虚拟环境
conda create -n torch_cuda11 python=3.6.3,这个 torch_cuda11是我自己定义的虚拟环境名字
另外,删除环境是conda remove -n envs_name –all
新建虚拟环境完成后会显示如下指令:

在Linux服务器上安装cuda11版pytorch

2.更新conda的源


conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

conda config --set show_channel_urls yes

3.更新pip的源

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

4.用pip安装pytorch
我用官网首页提供的指令有点小问题,安装的一直是1.4版本的pytorch+10.1版本的cuda,因此我在这里找了其他指令:https://pytorch.org/get-started/previous-versions/,用的是下面这个指令:

pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

安装完成后,用source activate torch_cuda11激活虚拟环境,然后进入python交互环境,然后输入 import torchtorch.__version__,如下图所示:

在Linux服务器上安装cuda11版pytorch

5.踩坑记录:
这次安装踩了特别多的坑,前前后后折腾了四五天,一些经验总结如下,不一定正确,但是确实是用了以下措施后才成功安装的:
1)anaconda3的版本不要太老,我的是4.3.21,应该是4.0版本里最新的了。5.0版本的anaconda3之前坑过我,所以没装5.0版本。
2)创建的虚拟环境时,python版本不要只写3.6,应该写具体点比如3.6.2,3.6.3等
3)如果官网首页提供的下载链接不正确,可以试试这个链接:https://pytorch.org/get-started/previous-versions/,这个链接是安装非最新版cuda的

Original: https://blog.csdn.net/yuanren201/article/details/123827899
Author: 阿袁的小园子
Title: 在Linux服务器上安装cuda11版pytorch

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/706903/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球