Ubuntu 安装 pytorch

使用 pytorch 进行深度学习训练。 这也是我不得不选择 linux 的原因。本文是我多次安装时遇到的问题的汇总（包括笔记本和台式机，初始版本是基于我的笔记本电脑），故很多点可能并不会遇到，略过即可。

系统：ubuntu 22.04
型号：Lenovo Yoga 14sIHU 2021
集显：Irix Xe (lspci | grep -i vga)
独显：GeForce MX450 (lspci | grep -i nvidia)

安装驱动

图形化界面安装

在Show Applications中搜索 Additional Drivers。

选择最其中一个显卡驱动进行安装。其中如果对驱动了解不多，可以选择最上面那个，或者多选几次。对于我的环境而言，我记得第一次安装成功时是选最上面那个。但第二次安装时发现选择最上面之后重启时候，没办法进入系统了，一直卡住。

在网上查找了部分资料了，发现 Yoga14s(MX450)-Ubuntu20.04-显卡驱动安装
和 联想小新pro14/yoga 14s 2021安装ubuntu/kali/manjaro问题排雷 都提到了禁用nouveau。于是利用如下命令禁用nouveau，根据 Ubuntu20.04 系统禁用自带的Nouveau驱动。

sudo vim /etc/modprobe.d/blacklist-nouveau.conf #该文件可能需要新建
1

在上述文件中添加如下语句。

blacklist nouveau
options nouveau modeset=0
1
2

更新initramfs

sudo update-initramfs -u
1

重启系统后，通过以下命令查看是否禁用，没有显示内容，则表示nouveau被成功禁用。

lsmod | grep nouveau
1

最后根据 ubuntu18.04 禁止nouveau 并安装NVIDIA显卡驱动 修改 /etc/default/grub 文件为
GRUB_CMDLINE_LINUX_DEFAULT="quiet splash acpi_osi=linux"。注意考虑到在文章 双系统安装 ubuntu 22.04 + win11（yoga 14s 2021） 中我对grub文件的修改。则综合就变成了"quiet splash acpi_osi=linux i915.enable_psr=0 i8042.dumbkbd"

最后试着换了下驱动，可以安装，同时安装之后能够正常进入系统。这里有一个小技巧就是，如果安装完之后有个restart按钮，那么便是正常（并没有经过严谨的检验）。另外，上述的工作只是充分条件，并不一定必要。

使用nvidia-smi检验是否安装成功。如果直接在additional drivers那一步restart后就能够正常显示结果，则就说明安装成功了，不需要包括禁用nouveau在内的任何其他步骤。

命令行安装

同时 Ubuntu安装nvidia驱动_小潇港哥的博客-CSDN博客 这篇文章提供了命令行安装驱动的方法。就两句+重启。

sudo ubuntu-drivers devices # 查找
sudo apt install nvidia-driver-470 # 安装
1
2

安装包安装

在某次安装中（没错，就是这么多次。。。），不知道是不是 apt 损坏了，或者卸载时候尝试各种方法给搞崩了，使用 sudo apt install nvidia-driver-535 出现如下报错。

The following packages have unmet dependencies:
nvidia-dkms-545 : Depends: dkms but it is not installable
E: Unable to correct problems, you have held broken packages.

最后没办法只能通过安装包安装，幸运的是这次安装成功了。唉