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