环境配置:Ubuntu+cuda+cudnn
驱动与安装前准备
(以下步骤也可参考官方适用于Linux的NVIDIA CUDA安装指南)
1. 验证自己的电脑是否有一个可以支持CUDA的GPU
lspci | grep -i nvidia
2. 安装NVIDIA驱动
这里可以先跳过,在之后安装cuda9.2的时候里面可以一起安装。
如果你是别的版本或者想自己尝试别的方法也可以试试下面的设置:
可以自己本地设置:
右上角>系统设置>软件和更新>附加驱动>选择NVIDIA那项>应用更改
或者下载官方的驱动自己安装
3. 验证自己的Linux版本是否支持 CUDA(Ubuntu 16.04没问题)
uname -m && cat /etc/*release
4. 验证系统是否安装了gcc
(注意所安装的gcc版本要满足官方CUDA所需gcc版本)
根据要求,我这里安装的是gcc5.4.0的版本。
gcc -v
显示如下:
若没安装的话,依次执行以下命令可以安装gcc5.4.0:
下载gcc5.4.0,然后在下载路径打开控制台执行以下命令:
tar -zxvf gcc-5.4.0.tar.gz
cd gcc-5.4.0 //进入解压后的gcc文件夹
./contrib/download_prerequisites
cd ..
mkdir gcc-build-5.4.0
cd gcc-build-5.4.0
../gcc-5.4.0/configure --enable-checking=release --enable-languages=c,c++ --disable-multilib
sudo make // 这个命令需要的时间比较久,请耐心
sudo make install
要输入密码的时候输入一下密码即可。(时间比较长,要有耐心)