Table of contents
CUDA+cuDNN安装
下载的CUDA版本必须要与所支持的驱动相对应
可以先将显卡更新至最新版本后再下载。下载安装NVIDIA显卡驱动
NVIDIA官网:https://www.nvidia.cn/geforce/drivers/
下载驱动程序,注册账号。
安装NVIDIA完成后,打开GEFORCE Experience,打开驱动程序,检查更新文件,若有,则下载更新文件

若输出结果为Ture,则安装成功,若为False,则失败
若输出为False 输入pip list,查看当前环境下有哪些包,若没有torch,则进入其官网进行下载
pytorch官网:https://pytorch.org/
进入官网下拉,找到对应torch版本
输入conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge
下载完成后,再一次输入pip list 查看是否存在torch包
若存在,再一次检测安装是否成功(输出为Ture)
若还是错误,尝试直接下载 .whl文件进行
下载地址:https://download.pytorch.org/whl/torch/
选择匹配自己python版本的torch、torchaudio和torchvision,cu115代表匹配cuda11.5版本,cp39表示匹配python3.9版本,win就表示windows
torch下载:https://download.pytorch.org/whl/torch/
torchaudio下载:https://download.pytorch.org/whl/torchaudio/
torchvision下载:https://download.pytorch.org/whl/torchvision/
三个文件下载后,放在平时安装的位置:C:\Users\lenovo,其后在cmd中将三个文件按顺序安装
cmd输入:pip install 文件名
安装成功后,文件夹中出现一下六个文件
再重新进入python环境,检查是否成功安装