win11下30系列显卡配置anaconda虚拟环境cuda,cudnn,pytorch

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

支持最高cuda版本是11.5

win11下30系列显卡配置anaconda虚拟环境cuda,cudnn,pytorch_第1张图片

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

win11下30系列显卡配置anaconda虚拟环境cuda,cudnn,pytorch_第2张图片

解压完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博客icon-default.png?t=M3K6https://blog.csdn.net/weixin_44791964/article/details/120668551

你可能感兴趣的:(pytorch,深度学习,vscode,windows)