CUDNN(CUDA深度神經網絡庫)是NVIDIA提供的一個用于深度學習的加速庫,它可以提供深度神經網絡的加速運算和訓練。以下是在Ubuntu系統上安裝CUDNN的指南:
下載CUDNN: 首先,你需要從NVIDIA官網上下載適用于你的CUDA版本的CUDNN。確保你下載的版本和你的CUDA版本兼容。
解壓CUDNN文件: 下載完成后,將CUDNN壓縮文件解壓到一個目錄中。
復制文件到CUDA目錄: 進入解壓后的CUDNN目錄,將其中的文件復制到你的CUDA安裝目錄中。通常情況下,CUDA安裝在/usr/local/cuda目錄下。
sudo cp -r include/* /usr/local/cuda/include/
sudo cp -r lib64/* /usr/local/cuda/lib64/
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda/bin:$PATH
保存修改后,執行以下命令使修改生效:
source ~/.bashrc
nvcc -V
如果成功安裝,會顯示CUDA和CUDNN的版本信息。
以上就是在Ubuntu系統上安裝CUDNN的簡單指南,希望對你有所幫助。