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

支持最高cuda版本是11.5

2.cuda cudnn 安装

（1）cuda

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

下载链接：

CUDA Toolkit 11.3 Update 1 Downloads | NVIDIA Developer/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 Developer/rdp/cudnn-archive

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

3.anaconda prompt终端：

conda create -n pytorch-gpu python=3.7conda activate pytorch-gpu#CUDA 11.3conda install pytorch==1.9.1 torchvision==0.10.1 torchaudio==0.9.1 cudatoolkit=11.3 -c pytorch -c conda-forgepip install tqdmpip install scipypip install opencv_pythonpip install matplotlibpip install tensorboard

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

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

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

注：

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

是因为setuptools版本太高，则：

pip uninstall setuptoolspip install setuptools==59.5.0

参考链接：

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