工程實踐_Ubuntu安裝CUDNN以及查看CUDA和CuDNN版本

查看CUDA和CuDNN版本

查看CUDA版本：

nvcc --version
或
nvcc -V

也可以通過安裝目錄查看：

cat /usr/local/cuda/version.txt

查看CuDNN版本：

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

CuDNN安裝

安裝了cuda之後，需要安裝cudnn加速。

我們可以去官網下載，進入官網後找到相應的資源，選擇選擇 cuDNN Library for Linux下載。

下載完成後我們進行解壓：

 tar -xzvf cudnn-10.1-linux-x64-v7.6.5.32.tgz

當面目錄會產生cuda文件夾，不進入文件夾，繼續輸入如下命令：

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*

這樣就完成了！

我們可以使用上面提到了命令進行檢查安裝是否成功：

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