pytorch 1.9.0+cuda 10.2安装

参考：CUDA（10.2）＋PyTorch安装加配置 详细完整教程_cuda10.2_Ang_go的博客-CSDN博客

参考以上博客安装好cuda10.2和cudnn8.7.0（我选的最新版本）

接下来安装pytorch 1.9.0

1、打开Anaconda prompt

2、创建虚拟环境：conda create -n pytorch_gpu python=3.9

3、激活环境：conda activate pytorch_gpu

4、下载pytorch1.9.0,用了清华源。

pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.9.0+cu102 torchvision==0.10.0+cu102 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

 安装结束后，在虚拟环境的终端输入python，之后输入以下代码：

import torch
torch.cuda.is_available()

返回True，表示环境可用。如下图：

 上述为pytorch gpu版本的安装过程。