学习笔记-机器学习-搭建环境-6:安装CUDNN

一、安装cuDNN

在CentOS上,NVIDIA驱动的RPM包依赖于一些外在的包(external packages)

# 从EPEL上面来安装dkms
yum install epel-release
yum install dkms

注册账号,下载cuDNN

网址:https://developer.nvidia.com/rdp/cudnn-download

# 解压solitairetheme8文件
tar -zxvf cudnn-9.0-linux-x64-v7.1.solitairetheme8

将下述文件拷贝到CUDA Toolkit文件夹,并更改权限

cp cuda/include/cudnn.h /usr/local/cuda/include
cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章