我的安装环境是python == 3.7.9。为了不出问题,故意选老一点的版本。使用pip安装,主要参考两篇博文:
cuda安装教程+cudnn安装教程_sinat_23619409的博客-优快云博客_cuda安装
配置显卡驱动、CUDA、cuDNN以及说明三者之间的关系_ZiLiangQin的博客-优快云博客_cudnn配置
还有官网的显卡配对关系:
Release Notes :: CUDA Toolkit Documentation (nvidia.com)
从pytorch的安装页面来看:
目前(2021年5月26日)为止,最高只支持到CUDA 11.1。但是官网的cuda已经显示11.3的版本。所以要先找到以前的版本,地方在这: