要在Ubuntu上配置cuDNN,首先需要確保已經安裝了NVIDIA的GPU驅動和CUDA。然后按照以下步驟進行配置:
下載cuDNN:首先需要登錄NVIDIA的開發者網站(https://developer.nvidia.com/cudnn),注冊并下載適用于你的CUDA版本的cuDNN壓縮包。
解壓縮cuDNN:解壓下載的壓縮包,并將其中的文件復制到CUDA的安裝目錄下。一般情況下,CUDA的安裝路徑為/usr/local/cuda/
。
設置環境變量:編輯~/.bashrc
文件,并添加以下幾行代碼:
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda
保存并退出文件,然后運行以下命令使配置生效:
source ~/.bashrc
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
如果成功顯示cuDNN的版本信息,則表示配置成功。
現在,您已經成功在Ubuntu上配置了cuDNN,可以開始在深度學習項目中使用它了。