参考Win10 下安装 CUDA Toolkit ( for PyTorch )
1、查看电脑的CUDA驱动版本,确定cuda toolkit的兼容版本。
最好是让cuda toolkit的版本低于cuda的驱动版本
在 CUDA Toolkit 历史版本 中,选择 CUDA 11.0.3 进行下载。
2、下载和安装 cuda 11.0
成功之后,C盘下面的路径
3、下载并安装与CUDA11.0版本兼容的CUDNN(为什么要安装cudNN)可以理解为是和cuda配套的东西,下载 cuDNN 需要注册,官网下载地址为 https://developer.nvidia.com/cudnn
将所有内容复制到 cuda 11.0 安装目录(在C盘里面),即可完成 cuDNN 的安装。
以上步骤,缺一不可。
现在,才正式开始安装Pytorch.
可以直接安装pytorch\torchvision\cuda toolkit\ torch audio