Ubuntu深度学习环境搭建

1.关于安装Ubuntu系统

Ubuntu系统安装的教程有很多，这里不做赘述，主要记录一下带有集成显卡和独立显卡安装Ubuntu系统出现问题如何解决。带有集显和独显的电脑安装Ubuntu系统需要进入BIOS关闭安全模式和禁用集成显卡，惠普工作站BIOS操作界面如下所示。

2. 安装显卡驱动准备工作

• 2.1安装依赖包(可选)

刚安装的系统需要更新包和安装一些依赖包，比如make,gcc等, 另外建议更换apt包管理器的源为国内源。
操作命令：

sudo apt update

检查可更新的包

sudo apt upgrade

更新可更新的包

sudo apt install make gcc

安装依赖包

2.2 禁用nouveau驱动(可选)

若有集成显卡，须禁用nouveau驱动。nouveau是第三方开发的英伟达驱动，与官方驱动不兼容，非常头疼的一个玩意儿。

sudo gedit /etc/modprobe.d/blacklist.conf

用可视化编辑器gedit打开配置文件

blacklist nouveau

options nouveau modeset=0

在文件最后方插入以上两行，将nouveau加入黑名单。

sudo update-initramfs -u

更新系统内核，然后重启（必须重启！）

lsmod | grep nouveau

若无输出则禁用nouveau驱动成功。

3. 安装英伟达显卡驱动

在英伟达的官网上查找你自己电脑的显卡型号然后下载相应的驱动。网址：http://www.nvidia.cn/page/home.html

sudo chmod  a+x NVIDIA-Linux-x86_64-396.18.run

给驱动run文件添加可执行权限

sudo ./NVIDIA-Linux-x86_64-396.18.run -no-x-check -no-nouveau-check -no-opengl-files

-no-x-check：安装驱动时关闭X服务
-no-nouveau-check：安装驱动时禁用nouveau
-no-opengl-files：只安装驱动文件，不安装OpenGL文件

以上三个参数是可选参数，可按照实际情况选择添加与否。然后执行以上命令安装驱动。

nvidia-smi

输入以上命令，若出现如下界面则驱动安装成功。

4. 关于Ubuntu内核版本与驱动版本不一致问题

使用nvidia-smi若出现NVIDIA-SMI has failed because it couldn’t communicate with the NVIDIA driver问题，是Ubuntu内核版本和英伟达显卡驱动版本不一致导致的，可使用dkms动态内核模块支持包解决。

sudo apt install dkms

安装dkms

sudo dkms install -m nvidia -v xx

使用dkms自动维护内核版本，xx是显卡驱动版本号，比如我安装的是460.77, xx就为460.77。

5. 安装cuda与pytorch

cuda可以在官网下载也可以在pytorch官网下载，个人认为在pytorch官网的cuda包应该是cuda官网的一个阉割版，这里采取才pytorch官网下载的更简单的方式。需要注意的是显卡、显卡驱动版本、cuda版本、pytorch版本是有对应关系，具体自行百度。打开pytorch官网，找到如下安装指导，根据需求自行安装。

参考博客：https://blog.csdn.net/yinwangde/article/details/89439648