安装pytorch-gpu全过程

1.安装miniconda

Miniconda是conda的精简版安装程序。 它是Anaconda的小型引导程序版本，仅包含conda，python，它们依赖的软件包以及少量其他有用的软件包，包括pip，zlib和其他一些软件包。 使用conda install command可以从Anaconda存储库中安装720+个其他conda软件包。
官网下载地址：miniconda官网
清华镜像下载地址：清华镜像下载
安装：
1.wget+地址

wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/Miniconda3-py38_4.8.3-Linux-x86_64.sh --no-check-certificate

2.安装

bash Miniconda3-py38_4.8.3-Linux-x86_64.sh

3.验证：首先重启终端，然后验证其版本号

conda --version

2.创建虚拟环境

安装python-3.10版本

conda create -n py310 python=3.10

3.安装pytorch-gpu版本

默认电脑或服务器已安装cuda和cudnn。
使用默认源下载，而不用清华镜像源下载，因为清华镜像源没有gpu版本的pytorch

# CUDA 10.2
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=10.2 -c pytorch

其他版本可以在pytorch官网中寻找语句
pytorch官网

安装全程都在命令行中运行。