學習筆記-機器學習-搭建環境-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*

 

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