深度学习脚本-GPU训练

一、查看使用电脑的NVIDIA版本信息

在cmd中输入nvidia-smi 首先要查找到nvidia-smi的路径信息
结果如下，版本号在红框

二、查看与NVIDIA 版本号对应的cuda版本

参考链接：

三、在CUDA官网进行下载

CUDA各版本下载链接：https://developer.nvidia.com/cuda-toolkit-archive
CUDA版本确定参考链接：https://blog.youkuaiyun.com/qq_43585377/article/details/114387345

四、查找与CUDA版本匹配的CUDNN版本

cudnn各版本下载链接：https://developer.nvidia.com/rdp/cudnn-archive

五、安装CUDA和cudnn

参考链接：https://blog.youkuaiyun.com/bingo_liu/article/details/103224730
注意：将cudnn解压之后，每个文件夹中对应的文件依次放入与cuda文件夹名相同的文件夹中

六、安装paddlepaddle-gpu

参考链接：https://www.paddlepaddle.org.cn/install/quick?docurl=/documentation/docs/zh/install/pip/windows-pip.html

七、pytorch安装

参考链接：https://pytorch.org/