1、显卡驱动安装
我在这里选用的ubuntu16系统,因为ubuntu16在ubuntu中算是经典系统,用户基数也大,各种环境基本上也能适配。系统在安装好系统后,进行系统软件源更新
sudo apt update
更新好系统源之后,在系统设置->软件和更新->附加驱动里选择NVIDIA驱动,因为我的显卡是GTX1060,所以在这里他给我推荐的是NVIDIA384驱动,点击应用之后等待它慢慢安装完成之后重启即可。
重启之后打开终端输入nvidia-smi查看显卡驱动是否安装成功
其他方法参考下面的博客,个人推荐第二种方法,博主讲得十分详细:
https://blog.csdn.net/u014682691/article/details/80605201
因为可能部分电脑安装完ubuntu系统后,系统不会检测出需要安装的驱动程序,所以上面链接中第二种方法是最为简单实用的。
2、CUDA安装
NVIDIA-linux-384对应的cuda版本是9.0,具体对应关系如下图
Linux系统下对应关系进入英伟达cuda专区下载对应版本cuda,我在这里下载的是cuda9.0,文件类型最好是要选择run file,基本不会出错
然后将cuda文件拷贝到HOME文件夹,按住control+alt+F1进入命令行界面,登陆上自己账号
输入命令sudo sh /home/username/cuda_9.0.176_384.81_linux.run进行安装
具体安装过程按照指引来进行就行,需要注意的是在安装cuda时系统也会默认给你安装显卡驱动,在这里我们在显卡驱动选项时选择no即可,其他项默认安装,安装路径选择默认路径
安装好cuda之后,按住control+alt+F7进入图形界面,打开终端要添加环境变量:sudo gedit ~/.bashrc
在最后一行添加
export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH
点击保存之后退出,输入source ~/.bashrc刷新一下环境,输入nvcc -V测试cuda是否安装成功
特别注意:安装完cuda一定先别重启!!!
安装cuda时会有一定概率重启时鼠标键盘不可用,去查了一下原因,是因为在安装cuda时可能会把xserver-xorg-input-all这个软件包给删掉,所以在重启系统之前一定要先把xserver-xorg-input-all重新安装回来,具体命令
sudo apt install xserver-xorg-input-all
需要什么依赖软件,对应安装好即可
3、安装cudnn
cudnn的安装我采用了最简单的方式deb文件安装,点击这里跳转到下载页,选择cuda9.0版本的cudnn
选择第456个文件选择ubuntu16对应的三个deb文件进行下载
下载cudnn是要注册nvidia账号的,注册完账号后加入小组即可下载文件
下载完成以后直接用sudo dpkg -i指令安装deb文件即可
验证方法:
cp -r /usr/src/cudnn_samples_v7/ $HOME
cd $HOME/cudnn_samples_v7/mnistCUDNN
make clean && make
./mnistCUDNN
如果输出结果为Test passed!即说明已经安装成功了
这时候重启就完成了!
网友评论