步骤:
1. python
sdk 3.9以上
2. 显卡驱动
可以使用驱动精灵 直接安装N 卡推荐
3. 安装机器学习套件CUDA
cuda 安装在PyTorch
需要根 PyTorch版本一致,我的 win-srv 最高支持 12.1
下载地址: https://developer.nvidia.com/cuda-toolkit-archive
https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_531.14_windows.exe
记录下上图的 cuda安装路径:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.1\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.1\lib
CUDA_HOME=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.1\bin
LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
PATH=%CUDA_HOME%\bin
验证CUDA套件
nvcc -V
4、配置cuDNN
通俗点讲,cuDNN就是CUDA toolkit的一个补丁,深度学习需要这个补丁才能使用API驱动GPU的CUDA
下载: cuDNN 9.1.1 Downloads | NVIDIA Developer
进官网,需要登陆NVIDIA账号,并且注册成开发者,完成后进入下载界面
解压,全选复制,黏贴到CUDA安装目录,全部"是"即可
直接覆盖
5. PyTorch 安装(挖矿等这是不需要的)
地址: https://pytorch.org/
选择 N卡 cuda 或者 CPU 平台支持
选择对应的平台 版本: 我这边旋转 12.1
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121