Linux服务器下安装gpu版pytorch(用anaconda)

一、首先，下载anaconda，用

wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-2022.10-Linux-x86_64.sh

其次，修改权限并运行安装

chmod +x Anaconda3-2022.10-Linux-x86_64.sh
./Anaconda3-2022.10-Linux-x86_64.sh

安装过程依此输入enter和yes。

二、安装好后，创建新的虚拟环境

conda create -n torch python=3.7
conda activate torch

三、安装pytorch

找到官网的相关命令

https://pytorch.org/

命令如下(两种)：

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

pip3 install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple

搞定！