本片博客的安装环境是ubuntu16.04系统和已经安装好的GPU驱动,GPU安装驱动请自行完成。
安装好驱动之后执命令查看安装的驱动版本:
sudo nvidia-smi
可以看到我的驱动是384.130版
1. 安装CUDA
cuda的下载
我这里选择cuda-8.0.44进行下载。这个是下载cuda的链接地址:https://developer.nvidia.com/cuda-80-download-archive
下载下来的文件放到**/home**目录下,大家也可以选择适合自己电脑的CUDA版本进行下载。
cuda的安装
执行下面的命令进行安装:
sudo sh cuda_8.0.44_linux.run
注意:在安装的时候首先要一直按enter键完成安装文档的阅读,然后选择 **accept ** 接下来会出现让你选择是否安装nvidia361驱动,因为前面驱动已经安装好了,所以这里要选择no,不安装默认驱动
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 361.62?#就是这里要选择no
cuda环境的配置
打开**~/.bashrc** 文件,并添加3句命令到文件的末尾。
sudo vim ~/.bashrc #打开.bashrc文件
export CUDA_HOME=/usr/local/cuda-8.0 #将export这3句添加到文件的末尾。
export PATH=$PATH:${
CUDA_HOME}/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:${
CUDA_HOME}/lib64
如下图所示:
source