Ubuntu18.04深度学习框架搭建

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

参考