安装适合CUDA版本的pytorch

如果pytorch版本与CUDA版本不匹配，torch会找不到CUDA，导致无法使用GPU加速。

解决方案

1. 查看当前pytorch与CUDA版本

python -c 'import torch;print(torch.__version__);print(torch.version.cuda)'

2. 查找并安装所需的pytorch版本

查找网址： Previous PyTorch Versions | PyTorch ，选择对应CUDA版本的与pytorch安装。

举个例子，为CUDA 11.1安装torch 1.9.1：

pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html

安装后再查看版本，检查是否安装成功。