在服务器上配置Pytorch

目录

1、安装Anaconda3

2、将Anaconda3的安装包上传至服务器 

3、在服务器上安装Anaconda

4、配置清华源镜像（加速后续下载）

5、在Anaconda中配置虚拟环境

5.1 创建虚拟环境

5.2 激活虚拟环境

5.3 Pytorch下载 

5.4使用whl(轮子文件)安装pytorch(强烈推荐！！)

6、在VScode中使用pytorch

1、安装Anaconda3

在官网https://www.anaconda.com/products/distribution#macos上找到合适Linux版的Anaconda安装包：

2、将Anaconda3的安装包上传至服务器 

可以使用Xftp 7这款软件直接对本地和服务器上的文件进行操作

 

只需把本地的文件直接拖拽到右侧服务器上的区域即可。 

3、在服务器上安装Anaconda

bash Anaconda3-2022.10-Linux-x86_64.sh

在Xshell上直接执行这个指令，然后一路Enter/yes完成安装操作 

在安装的过程中，它可能会要你指定安装的目录，如果当前文件夹空间比较有限，你可以自定义安装到一个空间比较大的目录中:

Anaconda3 will now be installed into this location:
/home/xxx/anaconda3

  - Press ENTER to confirm the location
  - Press CTRL-C to abort the installation
  - Or specify a different location below

[/home/xxx/anaconda3] >>> /raid/xxx/yyy/anaconda3
PREFIX=/raid/xxx/yyy/anaconda3

之后，你的anaconda就会被安装到/raid/xxx/yyy/anaconda3这个路径

注意：如果使用自定义的安装路径，可能需要后续配置一下服务器上面的一些配置文件，好让anaconda找到你安装的位置。具体的配置方法可以问问GPT。

4、配置清华源镜像（加速后续下载）

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --append channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/fastai/
conda config --append channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --append channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --set show_channel_urls yes

 默认下载镜像源设置好了以后，你在任何一个虚拟环境下载软件或者包都是从这个源下载的。当然你也可以不更改默认的下载源，而是每次下载包的时候都在命令后面添加镜像源地址（也没有很费事）。

5、在Anaconda中配置虚拟环境

5.1 创建虚拟环境

conda create -n research python=3.11

5.2 激活虚拟环境

conda activate research

5.3 Pytorch下载 

Pytorch官网：PyTorch 

 

可以用Pytorch官网上面提供的代码来进行安装，但是会非常慢，比较推荐的方法是使用下面的阿里云镜像来进行安装：

pip install torch torchvision -i http://mirrors.aliyun.com/pypi/simple/  --trusted-host mirrors.aliyun.com

在pytorch虚拟环境中执行上述语句，就会在pytorch虚拟环境中安装pytorch了。

5.4使用whl(轮子文件)安装pytorch(强烈推荐！！)

进入下面这个网址，pytorch的下载官网官网. (可以把cu后面的数字更改为你现在要安装的cuda的版本)

假设我现在想要安装torch2.4.1+cu121 这个版本(表示torch版本为2.4.1，GPU对应的版本是1.2.1)

https://download.pytorch.org/whl/cu121

在页面中找到torch, torchvision, torchaudio; 点击进入。

torch, torchvision, torchaudio的版本对应关系在 Previous PyTorch Versions | PyTorch 查阅。

找到你要安装的版本，要和python版本对应好

e.g. torch-2.4.1+cu121-cp311-cp311-linux_x86_64.whl 表示 torch版本2.4.1， cu版本12.1, python版本3.11

将轮子文件下载下来

然后上传到服务器上，切换到你的虚拟环境，使用pip install命令安装即可。

pip install torch-2.4.1+cu121-cp311-cp311-linux_x86_64.whl 

6、在VScode中使用pytorch

VScode中需要先安装Remote SSH插件，并建立链接，详见这篇文章使用VScode连接远程服务器的配置方法_vscode hostname user-优快云博客

因为服务器上面装了python，因此如果我们不做调整，那默认的会使用服务器上装好的python，而不是我们自己装的虚拟环境pytorch。在VScode中进行以下操作：

第一步、 Ctrl+shift+P

第二步、选择Python： Select Interpreter，选择你的pytorch虚拟环境中python解释器

在上图中，第一行(base)表示的当前的环境是base，使用conda activate pytorch就可以切换到pytorch这个环境中；

第二行的(pytorch)就表示当前环境是pytorch。

zht@daitao:~$ ：表示当前所在的目录