新手安装cuda和cudnn + pytorch记录

我的安装环境是python == 3.7.9。为了不出问题,故意选老一点的版本。使用pip安装,主要参考两篇博文:

cuda安装教程+cudnn安装教程_sinat_23619409的博客-CSDN博客_cuda安装

配置显卡驱动、CUDA、cuDNN以及说明三者之间的关系_ZiLiangQin的博客-CSDN博客_cudnn配置

还有官网的显卡配对关系:

Release Notes :: CUDA Toolkit Documentation (nvidia.com)

从pytorch的安装页面来看:

新手安装cuda和cudnn + pytorch记录

目前(2021年5月26日)为止,最高只支持到CUDA 11.1。但是官网的cuda已经显示11.3的版本。所以要先找到以前的版本,地方在这:

CUDA Toolkit 11.3 Update 1 Downloads | NVIDIA Developer

新手安装cuda和cudnn + pytorch记录

找到11.1的版本下载即可:

新手安装cuda和cudnn + pytorch记录

在下载之前,先检查一下自己显卡的驱动版本有没有问题,检查的地方在这:

新手安装cuda和cudnn + pytorch记录

新手安装cuda和cudnn + pytorch记录

检查一下:Release Notes :: CUDA Toolkit Documentation (nvidia.com) 是不是符合这个网址上说的驱动版本要求,如果不符合就更新一下。

然后接下来就是把下载好的安装和下载好cudnn的安装。注意事项请参考开头的两篇博客。

上一篇:linux服务器下命令行对anconda进行下载及其配置、conda使用、查看 cudnn& cuda


下一篇:Vue中keep-alive使用 include/exclude不起作用问题