centos8安装pytorch

1.下载miniconda

清华源

2. 安装miniconda

bash Anaconda3-2023.03-1-Linux-x86_64.sh

3. 换源

conda config --remove-key channels

conda config --show channels

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
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/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

conda config --set show_channel_urls yes

4. 新建虚拟环境

conda create -n env_name
conda activate env_name

5. 安装pytorch

官方（-c pytorch会导致使用官方源）

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

也可是不指定cuda版本，会自动下载合适版本

conda install pytorch torchvision torchaudio 

6. 检查

查看torch版本

import torch
print(torch.__version__)

查看cuda版本

print(torch.version.cuda)

查看cudnn版本

print(torch.backends.cudnn.version())

查看GPU可用

print(torch.cuda.is_available())