本文已参与「新人创作礼」活动,一起开启掘金创作之路。
查看cuda 版本
cat /usr/local/cuda/version.txt
查看cudnn 版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
查看显卡驱动版本
cat /proc/driver/nvidia/version
或输入nvidia-smi
cuda安装
安装地址:CUDA Toolkit Archive | NVIDIA Developer
下载对应的runfile文件,执行下面代码:
sudo sh cuda_10.0.130_410.48_linux.run
但是出现了问题,报错如下:
Installing the NVIDIA display driver... The driver installation is unable to locate the kernel source. Please make sure that the kernel source packages are installed and set up correctly. If you know that the kernel source packages are installed and set up correctly, you may pass the location of the kernel source with the '--kernel-source-path' flag.
此处解决方法为:安装 dkms
sudo apt install dkms
然后再次运行命令即可:
sudo sh cuda_10.0.130_410.48_linux.run
出现这种情况时,加sudo sh cuda_8.0.44_linux.run --tmpdir=/media/zxc/SOFTWARE/即可!
成功安装:
然后修改环境变量:
vim ~/.bashrc
最后添加上:
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.0/lib64
export PATH=$PATH:/usr/local/cuda-10.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.0
更新:
source ~/.bashrc
如果有多个版本的cuda怎么办呢?怎么切换呢?
此种安装方式,是将 /usr/local的cuda文件链接到了对应cuda版本的根目录。输入以下命令可以查看当前cuda文件的链接情况:
cd /usr/local/
stat cuda
随后会输出以下信息:
在 /usr/local下,我们可以安装多个版本的cuda,这里我列出我电脑内的cuda版本,有cuda9.0、cuda10.0等多个版本
命令行输入以下命令,断开cuda文件与cuda-10.1的符号链接:
sudo rm -rf cuda
命令行输入以下命令,建立cuda文件与cuda-9.0的符号链接:
sudo ln -s /usr/local/cuda-9.0 /usr/local/cuda
此时查看cuda版本,已经更改成功,更换其他版本同样操作即可:
- 驱动安装完成后,下载cuda安装包(developer.nvidia.com/cuda-toolki…)
- sudo ./cuda_8.0.61_375.26_linux.run
- 一直按ctrl+F,直到100%
- 其它的按部就班,但选择不安装驱动
- 出现***WARNING: Incomplete installation! This installation did not install the CUDA Driver. A driver of version at least 361.00 is required for CUDA 8.0 functionality to work时,不用管它,此时只要安装的驱动版本更高比它更高即可
- 将cuda添加到环境变量,在~/.bashrc文件中末尾加上 export PATH=/usr/local/cuda-8.0/bin{PATH:+:{PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64{LD_LIBRARY_PATH:+:{LD_LIBRARY_PATH}}
- 在NVIDIA_CUDA-8.0_Samples中编译相关samples,能编译通过则说明安装成功
nvidia驱动安装
安装驱动地址(包含早期地址):NVIDIA 驱动程序下载 - 高级搜索
sudo apt-get purge nvidia*
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-384 nvidia-settings
reboot
注:如果要是不行,可能需要在bios里将显卡先设置成CPU集卡 验证 1.输入nvidia-smi查看 2.prime-select query查看当前选用的显卡
cuda卸载
sudo apt autoremove cuda
# 若上一步删不干净,接着使用: sudo apt --purge remove "*cublas*" "cuda*"
# 在之前下载的文件目录下运行: sudo sh cuda_10.2.89_440.33.01_linux.run
cd /usr/local/cuda/bin
sudo ./uninstall_cuda_8.0.pl
cd /usr/local
sudo rm -rf cuda-8.0
cudnn安装
安装地址:cuDNN Archive | NVIDIA Developer
参考:docs.nvidia.com/deeplearnin…
cudnn在linux系统下安装
下载文件,解压后,在解压后的当前文件夹打开终端,以此执行下面三个代码安装cudnn:
sudo cp cuda/include/cudnn.h /usr/local/cuda-8.0/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-8.0/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda-8.0/lib64/libcudnn*
到此cuda与cudnn全部安装成功!
reference:[Linux下安装cuda和对应版本的cudnn水哥不水的博客-CSDN博客linux安装cuda和cudnn](
Installing from a Tar File
Navigate to your directory containing the cuDNN Tar file. Unzip the cuDNN package.
$ tar -xzvf cudnn-9.0-linux-x64-v7.tgz
Copy the following files into the CUDA Toolkit directory, and change the file permissions.
sudo cp cuda/lib64/libcudnn /usr/local/cuda/lib64* $ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
Installing from a Debian File
Navigate to your directory containing cuDNN Debian file. Install the runtime library, for example:
sudo dpkg -i libcudnn7_7.0.3.11-1+cuda9.0_amd64.deb
Install the developer library, for example:
sudo dpkg -i libcudnn7-devel_7.0.3.11-1+cuda9.0_amd64.deb
Install the code samples and the cuDNN Library User Guide, for example:
sudo dpkg -i libcudnn7-doc_7.0.3.11-1+cuda9.0_amd64.deb