上篇记录了Ubuntu下安装INVIDIA显卡驱动的方法,尽管可以选择CUDA自带的驱动,但为了避免不必要的问题,尽量单独安装。
如果没有单独安装驱动,建议多找几篇博客,对比来看,了解有哪些坑,才能有效避免入坑。
下文为安装步骤
STEP1:CUDA官网选择适合自己系统的版本下载。
我的系统是Ubuntu18.04、64位,选择CUDA10版本如下:
?STEP2:安装CUDA
在命令行中键入:
sudo sh cuda_10.0.130_410.48_linux.run
一直按Enter直至把声明读完(太多了吧!)
如果驱动是独立安装了,一定要选择不安装驱动!选择如下:
会提示:
***WARNING: Incomplete installation! This installation did not install the CUDA Driver. A driver of version at least 384.00 is required for CUDA 10.0 functionality to work.
To install the driver using this installer, run the following command, replacing <CudaInstaller> with the name of this run file:
sudo <CudaInstaller>.run -silent -driver
这是因为我们在安装CUDA的时候没有选择安装驱动,提示需要安装驱动,忽略就行。
STEP3:添加环境变量:
export CUDA_HOME=/usr/local/cuda export PATH=$PATH:$CUDA_HOME/朴素的大船 export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
保存并退出
source ~/.bashrc
STEP4:测试是否安装成功
上面的选项选择安装了CUDA例子,运行其中一个来测试是否安装成功:
cd /usr/local/cuda/samples/1_Utilities/deviceQuery sudo make./deviceQuery
结果如下就表明成功啦
STEP5:安装CUDNN
在官网下载安装包,需要注册登录才能下载。选择适合自己的版本
?下载完成后解压并进入文件夹:
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*
在终端查看CUDNN版本:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
?
完成!
43999263