英伟达官网默认都是最新的,想下载历史版本一时不好找,记录一下地址
CUDA Toolkit Archive | NVIDIA Developer
以上是cuda文件,如果是安装cuDnn用于神经网络训练,下载地址是
cuDNN Archive | NVIDIA Developer
如果是纯粹的显卡驱动,下载地址是
安装cuda方法在下载页面有提示,一步步照着做就行,但是cuDNN的安装方法下载页面并没有提示,文档地址在这里
Installation Guide :: NVIDIA cuDNN Documentation
以上3种关系需要自己搞清楚,否则后果自负
在使用tensorflow过程中,有可能会提示缺少zlibdll,需要单独下载,然后放到cuda安装目录下,
下载地址为:ZLIB DLL Home Page — ZLIB DLL 主页 (winimage.com)
_________________________________________________________________
对于torch的安装也是一个头疼的问题,需要与cuda版本对应,如果是gpu版本,需要指定版本号,比如pip install torch==2.0.0+cu117 torchvision==0.15.1+cu117 torchaudio==2.0.1+cu117 -f https://download.pytorch.org/whl/torch_stable.html
可以参照一下帖子:GPU版本pytorch(Cuda12.1)清华源快速安装一步一步教!小白教学~_清华源安装torch-CSDN博客