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")
结果如下:
至此安装成功