记录安装pytorch-gpu

1. 查看cuda版本

nvidia-smi

安装的cuda不得大于该版本

2. 安装torch、torchvision、torchaudio三个组件

2.1 下载 torch

下载网址
首先选择torch，ctrl + F 搜索 [cu118-cp38-cp38-win] 这里cu118 是下载的 CUDA 11.8 版本，cp38-cp38 是说 Python 版本是 3.8。

2.2 查看对应版本的torchvision、torchaudio

查看网址
ctrl + F 搜索 [pip install torch==2.0.0] 并且对应cuda为11.8。

因此torchvision需要安装0.15.1版本，torchaudio需要安装2.0.1版本。

2.3 下载torchvision、torchaudio

在之前的网址中选择 torchvision，ctrl + F 搜索 [cu118-cp38-cp38-win]，下载版本为 0.15.1 的。

选择 torchaudio，ctrl + F 搜索 [cu118-cp38-cp38-win]，下载版本为 2.0.1 的。

2.4 创建python虚拟环境并激活

conda create -n your_env_name python=x.x

conda activate your_env_name

2.5 安装

按照三个文件的存放路径进行安装，例如

对应的命令应该是

pip install C:\Users\sybs\Downloads\torch-2.0.0+cu118-cp38-cp38-win_amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install C:\Users\sybs\Downloads\torchaudio-2.0.1+cu118-cp38-cp38-win_amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install C:\Users\sybs\Downloads\torchvision-0.15.1+cu118-cp38-cp38-win_amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple

安装过程耐心等待。

安装结束后需要测试是否成功安装gpu版本的pytorch，这里依旧在创建的环境下输入python，进入python编程环境后输入 import torch 回车后输入 torch.cuda.is_available() 。如果返回True则安装成功。

其他

安装 cv2、tqdm、matplotlib、captum、skimage：

conda install --channel https://conda.anaconda.org/menpo opencv
conda install tqdm
conda install matplotlib
conda install captum
pip install scikit-image

删除环境：

conda env remove -n env_name

参考：
pytorch gpu版安装（普适稳妥，亲测有效） - 知乎 (zhihu.com)
# annaconda 安装 opencv（cv2）