一、查看使用电脑的NVIDIA版本信息
在cmd中输入nvidia-smi 首先要查找到nvidia-smi的路径信息
结果如下,版本号在红框
二、查看与NVIDIA 版本号对应的cuda版本
参考链接:
三、在CUDA官网进行下载
CUDA各版本下载链接:https://developer.nvidia.com/cuda-toolkit-archive
CUDA版本确定参考链接:https://blog.csdn.net/qq_43585377/article/details/114387345
四、查找与CUDA版本匹配的CUDNN版本
cudnn各版本下载链接:https://developer.nvidia.com/rdp/cudnn-archive
五、安装CUDA和cudnn
参考链接:https://blog.csdn.net/bingo_liu/article/details/103224730
注意:将cudnn解压之后,每个文件夹中对应的文件依次放入与cuda文件夹名相同的文件夹中
六、安装paddlepaddle-gpu
七、pytorch安装
参考链接:https://pytorch.org/