环境:
Win11 22H2
需要的安装包:
Anaconda3-2021.05-Windows-x86_64.exe
Python 3.11.(pytorch 2.0目前推荐的Python 版本为3.8-3.11)
pycharm-professional-2021.2.1.exe
CUDA12.1 与 CUDNN V8.9.5
pytorch 2.1
选择性安装OpenCV库
一、安装CUDA12.1与CUDNN V8.9.5
- CUDA:是显卡厂商NVIDIA推出的运算平台。 CUDA是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。
- CUDNN:NVIDIA cuDNN是用于深度神经网络的GPU加速库。它强调性能、易用性和低内存开销。
1、安装CUDA12.1
(1)安装前检查GPU可支持的最高CUDA版本
由下图可以看出我的GPU支持的最高CUDA版本为“12.2.79”,所以我安装CUDA12.1是完全可以的。
(2)
2、CUDNN V8.9.5
二、安装Anaconda+Python 3.11.
三、安装pycharm-professional-2021.2.1.exe
四、检查运行环境
标签:Pytorch2.1,3.11,CUDA12.1,CUDNN,CUDA,Anaconda,GPU,安装 From: https://www.cnblogs.com/qq2806933146xiaobai/p/17753265.html