cuda和cuDNN的安装

进行深度学习需要安装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

你可能感兴趣的:(电脑小问题,python,深度学习,人工智能)