pytorch Linux环境安装与使用(CPU 与GPU版本)

假定系统已经安装好了Anaconda并配置了python环境，pytorch的安装只需要去pytorch的官方网站查看对应的命令输入即可。

pytorch的官网可点击这里

对于Linux系统CPU版本可以使用以下conda或者pip命令安装(1.11.0版本，其他版本命令可参考官网)：

#CPU Only
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cpuonly -c pytorch
pip install torch==1.11.0+cpu torchvision==0.12.0+cpu torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cpu

对于Linux系统的GPU版本可以使用以下conda或者pip命令安装(1.11.0版本，其他版本命令可参考官网)：

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

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

# CUDA 11.3
pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113

# CUDA 10.2
pip install torch==1.11.0+cu102 torchvision==0.12.0+cu102 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu102

安装完成后，可打开python命令行，检验安装是否成功：

import torch
torch.__version__

如果显示版本号，则安装成功。