在linux远程服务器配置pytorch-gpu

一般租远程服务器的话，都是租linux系统的

本文就以在2080ti显卡服务器上安装pytorch-gpu版本为例。

首先输入python

可以看到这个服务器是已经安装好anaconda了，我们直接使用conda命令进行安装即可

conda info -e

 可以看到我之前安装的tensorflow1.13版本gpu环境，想直接怎么安装tf1.13.1版本gpu的可以看我之前写的博客，传送门

首先创建一个虚拟环境，我这里安装pytorch1.7.1版本的，python为3.7版本，大家也可以根据自己需求，安装相应的版本，命名也可以随意，比如pytorch1.7.1这个换成py1.7是不是简短呢

conda create -n pytorch1.7.1 python=3.7

安装完环境通过conda 进入到这个环境里面

conda activate pytorch1.7.1

然后安装的话根据pytorch官网的安装命令，一行代码就安装成功了，大家根据自己的安装版本，选择相应的命令安装

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch

最后输入python

import torch
print(torch.cuda.is_available())

完成安装