STEP1:单独安装驱动
终端运行:
sudo apt update
sudo ubuntu-drivers autoinstall
STEP2:CUDA官网选择适合自己系统的版本下载。
我的系统是Ubuntu18.04、64位,选择CUDA10版本如下:
注意:
cuda10.0版本,下载runfile文件,不要deb文件
STEP3:安装CUDA
在文件所在的文件中打开终端,在命令行中键入:
sudo sh cuda文件全名 -overrider
一直按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 with the name of this run file:
sudo .run -silent -driver
这是因为我们在安装CUDA的时候没有选择安装驱动,提示需要安装驱动,忽略就行。
STEP4:添加环境变量:
进入一下路径
sudo gedit ~/.bashrc
将以下变量添加到文件的最后一行后边
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
保存并退出
source ~/.bashrc
STEP4:测试是否安装成功
依次输一下:
nvcc -V
nvidia-smi
which pip
which python
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/include/cudnn.h
也可以参考视频:
[复制这段内容后打开百度网盘手机App,操作更方便哦]
链接: 提取码:n7wF