linux安裝cuda cudnn


參考博客:
https://blog.csdn.net/u014595019/article/details/53732015

https://blog.csdn.net/wanzhen4330/article/details/81699769

https://blog.csdn.net/auh_alias/article/details/80690556

一 cuda安裝

https://developer.nvidia.com/cuda-downloads

1下載run類型,安裝官網代碼提示:

在這裏插入圖片描述
accept
n(不要安裝driver,如果安了)取消x

2聲明環境變量

安裝完畢後
export PATH=/usr/local/cuda/binKaTeX parse error: Expected '}', got 'EOF' at end of input: {PATH:+:{PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64KaTeX parse error: Expected '}', got 'EOF' at end of input: …LIBRARY_PATH:+:{LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda
source ~/.bashrc 使環境變量生效
nvcc -V 查看安裝成功!

二 cudnn

https://developer.nvidia.com/rdp/cudnn-archive

1 安裝cudnn

解壓下載的文件,可以看到cuda文件夾,在當前目錄打開終端,執行如下命令:
tar -xvf cudnn-10.1-linux-x64-v7.6.3.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
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

2 查看cudnn版本

在終端輸入
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章