我的安装环境是python == 3.7.9。为了不出问题,故意选老一点的版本。使用pip安装,主要参考两篇博文:
cuda安装教程+cudnn安装教程_sinat_23619409的博客-CSDN博客_cuda安装
配置显卡驱动、CUDA、cuDNN以及说明三者之间的关系_ZiLiangQin的博客-CSDN博客_cudnn配置
还有官网的显卡配对关系:
Release Notes :: CUDA Toolkit Documentation (nvidia.com)
从pytorch的安装页面来看:
目前(2021年5月26日)为止,最高只支持到CUDA 11.1。但是官网的cuda已经显示11.3的版本。所以要先找到以前的版本,地方在这:
CUDA Toolkit 11.3 Update 1 Downloads | NVIDIA Developer
找到11.1的版本下载即可:
在下载之前,先检查一下自己显卡的驱动版本有没有问题,检查的地方在这:
检查一下:Release Notes :: CUDA Toolkit Documentation (nvidia.com) 是不是符合这个网址上说的驱动版本要求,如果不符合就更新一下。
然后接下来就是把下载好的安装和下载好cudnn的安装。注意事项请参考开头的两篇博客。