进行深度学习需要安装cuda进行gpu加速

主要需要安装cudatoolkit和cuDNN

cuda

官网安装路径：
https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64

在其中找到自己电脑GPU能够适配的类型（自己电脑的GPU驱动程序版本要和与cuda版本对应）
对应关系在官网中能够找到：
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

检测cuda是否安装成功：
进入cmd窗口，输入nvcc --version
或者set cuda查看cuda设置的环境变量

cuDNN

cuDNN下载的官网路径：
https://developer.nvidia.com/rdp/cudnn-download

找到和cuda版本对应的即可
下载完毕后将其粘贴到cuda的安装路径

检查cuDNN是否安装成功：
使用CUDA内置的deviceQuery.exe 和 bandwithTest.exe
cd到安装目录下的 …\extras\demo_suite,然后分别执行bandwidthTest.exe和deviceQuery.exe
得到result=pass即为成功

以上过程注意环境变量的配置，cuda安装路径下的bin、include、lib、libnvvp均需加入环境变量path

参考：https://blog.csdn.net/weixin_43848614/article/details/117221384