前言
ubuntu22.04安装nvidia驱动、cuda以及pytorch记录,主要是为了纠正网上很多文章的错误,避免走弯路。
我是在笔记本电脑测试的,安装了win11和ubuntu22.04双系统。
笔记本配置是12代i7、16G内存、3060 6G 显卡。安装ubuntu的时候就会出现黑屏无法启动的问题,这是显卡驱动的缘故,在grub提示下按e,然后添加nomodeset就可以进入系统了,这个按网上的文章叙述操作,基本都是可以的。
此外,装好系统之后还需要装个anaconda或者miniconda,这个按网上说的操作就行了,我装的miniconda。
一、安装nvidia显卡驱动
这个有很多方法,我是采用比较懒的方式,直接打开附加驱动,然后选择一个nvidia专用驱动安装,我选的是闭源的,目前版本是535和545,我安装的是545。装好之后重启即可。
二、安装cuda-toolkit
装好nvdia显卡驱动后,打开终端运行
nvidia-smi
可以看到如下信息: