Skip to content

Instantly share code, notes, and snippets.

@nanmi
Last active December 30, 2021 09:29
Show Gist options
  • Save nanmi/1db7bcd03ef34641dfcc7655f7bc85ca to your computer and use it in GitHub Desktop.
Save nanmi/1db7bcd03ef34641dfcc7655f7bc85ca to your computer and use it in GitHub Desktop.
Install CUDA & cuDNN

安装CUDA和cuDNN

安装CUDA

link: https://developer.nvidia.com/cuda-toolkit-archive

$ sudo chmod +x cuda_{xx.x.xxx_xxx.xx}_linux.run
$ sudo sh cuda_10.2.89_440.33.01_linux.run
Do you accept the previously read EULA?
accept/decline/quit: accept

#是否安装显卡驱动,选择否
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 {xxx.xx}?
(y)es/(n)o/(q)uit: n

#是否安装cuda
Install the CUDA 10.0 Toolkit?
(y)es/(n)o/(q)uit: y

#输入cuda的安装路径,默认
Enter Toolkit Location
[ default is /usr/local/cuda-10.0 ]: 

#是否建立/usr/local/cuda -> /usr/local/cuda-10.x的软连接
Do you want to install a symbolic link at /usr/local/cuda?
(y)es/(n)o/(q)uit: y

#是否安装CUDA示例
Install the CUDA 10.0 Samples?
(y)es/(n)o/(q)uit: n
增加环境变量
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
查看cuda版本
$ nvcc -V

安装cuDNN

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

$ tar -zxvf cudnn-{CUDA version i.e. 10.0}-linux-x64-v7.6.5.32.tgz
sudo cp cuda/include/cudnn* /usr/local/cuda/include
sudo cp -d cuda/lib64/libcudnn* /usr/local/cuda/lib64
查看cuDNN版本
$ vim /usr/local/cuda/include/cudnn_version.h
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment