Linux系統下CUDA和cuDNN環境配置

Akmpfen發表於2023-05-12
LinuxDNN

本人配置環境：linux伺服器，ubantu18，顯示卡驅動11.0，安裝CUDA11.0和cuDNN8.2.1。

一、安裝CUDA11.0

1、先多找幾篇部落格，瞭解大概的流程，避免踩坑。

2、官網下載，複製官網命令到伺服器下載。https://developer.nvidia.com/cuda-toolkit-archive

 3、輸入命令，再按i鍵，進入vim的編輯模式。

sudo vim ~/.bashrc

4、在末尾新增如下兩行，注意裡面的路徑要換成自己的版本，再按esc鍵退出vim編輯模式，並輸入  ：wq  儲存並退出。

export PATH=/usr/local/cuda-11.0/bin${PATH:+:${PATH}}

 export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64\
 ${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

source ~/.bashrc

5、以下命令啟用配置，並使用nvcc -V檢視配置效果，如圖表示成功。

source ~/.bashrc

 二、安裝cuDNN8.2.1

1、官網下載和CUDA對應的版本 https://developer.nvidia.com/rdp/cudnn-archive

2、解壓檔案

tar -xzvf cudnn-11.3-linux-x64-v8.2.1.32.tgz

3、執行如下四條命令，注意換成自己的版本。

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

4、使用如下命令檢視cuDNN的版本號

cat /usr/local/cuda-11.0/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

至此安裝完成。

相關文章