在Ubuntu中安裝cuDNN(CUDA Deep Neural Network library)可以按照以下步驟進行:
首先,需要去NVIDIA官網注冊并下載cuDNN的壓縮包,地址為:https://developer.nvidia.com/cudnn
下載完成后,將壓縮包解壓縮到指定目錄,例如解壓到/usr/local
目錄下:
tar -xzvf cudnn-10.0-linux-x64-v7.6.5.32.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
注意:這里的cudnn-10.0-linux-x64-v7.6.5.32.tgz
是根據你下載的具體版本號和系統架構來確定的,需要根據實際情況進行更改。
.bashrc
或.zshrc
等文件中添加如下路徑:export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
source ~/.bashrc
完成以上步驟后,cuDNN就成功安裝到系統中了。可以檢查是否安裝成功,以及cuDNN的版本等信息。