CUDA(Compute Unified Device Architecture),是显卡厂商 NVIDIA 推出的运算平台。 其并行计算架构使GPU能够解决复杂的计算问题。
cuDNN是基于CUDA的深度学习GPU加速库,有了它才能在GPU上完成深度学习的计算。
把CUDA比喻成一个工作台,cuDANN就是工作台上的工具零件,想要在CUDA上运行深度神经网络,就需要先安装cuDANN。就像手术台和手术刀的关系,如果只有手术台,而没有手术刀等相关工具,是无法进行手术的。
一、安装CUDA和cuDNN
CUDA下载地址:https://developer.nvidia.com/cuda-toolkit-archive
cuDNN下载地址:https://developer.nvidia.com/cudnn
CUDA和cuDNN的安装:https://blog.csdn.net/u010618587/article/details/82940528
https://blog.csdn.net/sinat_23619409/article/details/84202651
安装过程中的注意事项:
1、安装的CUDA版本要与电脑上NVIDIA匹配。
2、安装的cuDNN版本要与先前装的CUDA版本匹配。
安装10.1版本的CUDA成功,如下图所示(桌面-->右键-->NVIDIA控制面板):
NVIDIA显卡CentOS下安装CUDA教程:https://eipi10.cn/deep-learning/2019/11/28/centos_cuda_cudnn/
网友评论