1.电脑配置：win11+3060显卡

支持最高cuda版本是11.5

2.cuda cudnn 安装

（1）cuda

截止2022.5.6 pytorch支持的最新版cuda是11.3，故安装的cuda版本：

下载链接：

CUDA Toolkit 11.3 Update 1 Downloads | NVIDIA Developerhttps://developer.nvidia.com/cuda-11-3-1-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exe_local

（cuda安装选择自定义，其余顺着指示点下一步即可完成安装）

（2）cudnn

cudnn安装对应版本：

 下载链接：cuDNN Archive | NVIDIA Developerhttps://developer.nvidia.com/rdp/cudnn-archive

解压完cudnn压缩包后将压缩包内所有文件复制到如下目录：

3.anaconda prompt终端：

conda create -n pytorch-gpu python=3.7

conda activate pytorch-gpu

 #CUDA 11.3
conda install pytorch==1.9.1 torchvision==0.10.1 torchaudio==0.9.1 cudatoolkit=11.3 -c pytorch -c conda-forge

pip install tqdm

pip install scipy

pip install opencv_python

pip install matplotlib

pip install tensorboard

 查看pytorch和cuda对应版本的链接如下：

Previous PyTorch Versions | PyTorch​​​​​https://pytorch.org/get-started/previous-versions/很多依赖包都自动安装了： 

最终完成安装，代码运行正常。

---

注：

运行代码时报错AttributeError: module ‘setuptools._distutils‘ has no attribute ‘version‘

是因为setuptools版本太高，则：

pip uninstall setuptools
pip install setuptools==59.5.0

---

参考链接：

(1条消息) 深度学习环境配置8——（30系显卡）windows下的torch==1.7.1环境配置_Bubbliiiing的博客-CSDN博客https://blog.csdn.net/weixin_44791964/article/details/120668551