如何安装gpu版本的torchvision

安装 GPU 版本的 torchvision 有以下几个步骤：

1. 确认你的电脑上已经安装了 CUDA 和 cuDNN，可以通过 `nvcc --version` 和 `cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2` 命令来检查是否已经安装。

2. 创建一个新的虚拟环境：`conda create -n env_name python=3.7`。

3. 激活虚拟环境：`conda activate env_name`。

4. 安装 pytorch-gpu：`conda install pytorch torchvision torchaudio cudatoolkit=xx.x -c pytorch`（注意将 `xx.x` 替换为你安装的 CUDA 版本）。

5. 安装 torchvision：`pip install torchvision`。

这样就安装好了 GPU 版本的 torchvision。