【linux】一文解决服务器pytorch安装问题

【linux】一文解决服务器pytorch安装问题

创建环境，h_scope是我的环境名

conda create -n h_scope python=3.8

此处输入y，继续安装

y

source activate h_scope

激活环境，出现(h_scope)则表示已经进入自己创建的环境中了，接下来在该环境中安装pytorch

接下来要安装对应的pytorch版本
pytorch官网链接
版本对应链接（不知道python与pytorch版本对应的可以看一下这个链接）

点击上述红色方框，进入历史版本，找到自己想要安装的版本，因为我自己的需求是要安装cuda11.0版本，对应torch1.7.1，所以我选择下述红色方框的版本进行安装，大家根据自己需求安装

复制到终端，回车运行安装

pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

显示Successfully则表示安装成功，可以进一步检验一下是否真的安装成功，进入python

python
import torch
torch.cuda.is_available

返回是True则表示安装成功，检验无误
退出python，可以输入以下命令quit()退出，也可以直接摁住ctrl+D退出

quit()

查看安装的包情况

conda list

至此安装成功
因为我要的是cuda11.0的版本，为了进一步确认，我还可以在再打印一下版本信息

print(torch.version.cuda)

看完有帮助的话记得点赞+收藏+关注，谢谢大家！