首先声明:因为CUDA和CuDnn和pytorch都是在外网,所以进入他们的网站可能需要梯子,请自行搭梯子进入,博主这里提供不了梯子,为各位说一声抱歉。
1.CUDA的官网:CUDA Toolkit Archive | NVIDIA Developer
2.CuDnn的官网:https://developer.nvidia.com/rdp/cudnn-archive
3.Pytorch的官网:PyTorch
注意事项:
CUDA和Cudnn以及Pytorch的版本都要对应上,但是它们下载的顺序无关紧要,先下载哪个都可以,但是在安装CUDA和CuDnn之前先看一下Pytorch支持CUDA的版本是多少,避免CUDA版本过高而导致不匹配。在下载CUDA和CuDnn之前是需要登录NVIDIA的账号, 如果没有请自行注册。
在下载Pytorch的时候使用conda,pip或者其他下载的时候都有可能出现失败的机率。在国内的话最好添加镜像源下载以确保不会出现失败的情况。
ok,接下来开始
先确定自己的NVIDIA显卡所支持的CUDA版本是多少
有两个查看的办法