Linux服务器安装pytorch & 踩坑记录

第一部分 环境管理

步骤1 新建pytorch虚拟环境

conda create -n ENV_NAME python=3.x

注意：-n与–name等同 -n后是环境名称
严重注意：一定要预先确定你需要的python版本，因为不同的版本的python可pip安装的包都是不同的
建议是不要装最新的python，3.6或者3.7都可以

列出所有环境名称 确认新环境创建成功

conda info --envs

步骤2 切换到新创建的环境中

conda activate pytorch

第二部分 安装pytorch包

步骤0 查看服务器上的cuda版本

非常重要！为什么说非常重要呢，因为如果你瞎装了最新版的pytorch，和服务器上的已有的cuda版本不匹配的话，会出现严重的事故，那就是cuda.is_available()==False，直接利用不上gpu哦（这错误我犯多少次了呢 不记得了）

查看方式如下：

cat /usr/local/cuda/version.txt

或者

nvcc -V

也可以使用nvidia-smi查看（右上角）

注意：两者不匹配时，以前者为准

步骤1 安装pytorch包

首先，访问pytorch官网链接: pytorch官网.

查看与自己的cuda和python版本匹配的pytorch版本

查看previous pytorch versions

比如我新建的环境python版本是3.6，cuda版本是10.1，对照表格，我选择pytorch的版本是1.7.0

执行其命令即可

conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=10.1 -c pytorch

注意：至于torchvision和torchaudio不用可以不用管，但是cudatoolkit是一定要装的，否则还是会无法利用gpu

步骤2 检查是否安装成功

先conda list检查一下版本是否对应

而后可以import torch一下确认是否可以使用cuda

import torch 
if torch.cuda.is_available():
    print("cuda")
else:
    print("something wrong")

结果如下：

至此安装成功