Windows 11+RTX3050Ti安装CUDA、cuDNN与Pytorch的正确版本

实验环境为RTX 3050Ti的Windows 11。经实验,完全没问题的版本是:

1. CUDA Toolkit 11.1.1

CUDA Toolkit 11.1 Update 1 Downloads | NVIDIA Developericon-default.png?t=LA92https://developer.nvidia.cn/cuda-11.1.1-download-archive

依次选择:Windows-x86_64-10-exe(local)

2. ​cuDNN v8.1.1 (Feburary 26th, 2021), for CUDA 11.0,11.1 and 11.2​

cuDNN Archive | NVIDIA Developericon-default.png?t=LA92https://developer.nvidia.cn/rdp/cudnn-archive

3. Pytorch V1.8.0

安装指令为:

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

如果这篇博客对你有帮助,请给博主点个赞鼓励一下吧!

你可能感兴趣的:(报错与解决,pytorch,深度学习,cuda,cudnn)