ubuntu20.04安装pytorch深度学习环境(GPU RTX3060)
1. 安装显卡驱动
1.1. 下载驱动
1. 查看显卡型号
lspci | grep -i nvidia
由于驱动问题无法显示显卡的具体型号，在下面的网址输入显卡代号:2504来进行查询
http://pci-ids.ucw.cz/mods/PC/10de?action=help?help=pci
下面是我的显卡型号
2.下载驱动
去英伟达官网下载对应版本的显卡驱动。
驱动下载地址
1.2. 安装步骤
1. 禁用主板的Secure Boot
因为主板内置的公钥只有微软的，在ubuntu下安装Nvidia驱动的时候，会让你签名认证，否则安装不了。为了解决这个问题，进入BIOS ：设置Secure Boot 为disable
2.卸载原有的显卡驱动
sudo apt-get remove nvidia-*
sudo apt-get autoremove如果是runfile安装，使用下面的命令进行卸载
sudo chmod +x *.run
sudo ./NVIDIA-Linux-x86_64*.run --uninstall3. 禁用nouveau驱动
编辑blacklist.conf文件
sudo vim /etc/modprobe.d/blacklist.conf 在文本的最后添加:
blacklist nouveau
options nouveau modeset=0然后更新内核配置
sudo update-initramfs -u重启系统，执行下面命令 ，如果没有屏幕输出，说明nouveau禁用成功!
lsmod | grep nouveau4. 进入到字符界面
使用 Ctrl + Alt + F1 进入到字符界面(F1不行的话，可以尝试F3)， 输入账号密码登录。账号和密码为安装系统时设置的用户名和密码。
5.禁用X-window服务(命令行输入)
sudo sevice lightdm stop输入密码后等待服务停止
注意：
执行命令的时候如果报错：Failed to stop lightdm.service: Unit lightdm.service not loaded. 这可能是因为lightdm没有安装，安装lightdm命令
sudo apt-get install lightdm6. 安装驱动
sudo ./NVIDIA-Linux-x86_64-470.94.run -no-x-check -no-nouveau-check -no-opengl-files参数说明：
a. -no-opengl-files:：表示只安装驱动文件，不安装OpenGL。这个参数不可省略，否则导致登录界面死循环。
b. -no-x-check ：表示安装驱动时不检查X服务，非必须。
c. -no-nouveau-check：表示安装驱动时不检查nouveau，非必须。
注意：
如果提示, ERROR: Unable to find the development tool `cc` in your path; please make
sure that you have the package 'gcc' installed. If gcc is installed
on your system, then please check that `cc` is in your PATH.
解决办法： 安装gcc
sudo apt-get update
sudo apt-get install gcc
7. 根据提示完成操作（不同电脑提示略有不同，按照提示一步步操作即可）
（1）许可信息，选择左侧Accept（接受）
（2）提示有冲突： The distribution-provided pre-install script failed! Are you sure you
want to continue? 这个信息不必理会，选择左侧Continue installation
（3）开启构建NVIDIA核心模块。
（4）提示32位兼容库？我选侧的yes.
（5） 提示信息，回车
(6)启动X服务运行NVIDIA配置文件，自动更新X配置文件，选择左侧yes。
(7) 提示信息，安装成功.
8. 验证是否安装成功
9. 恢复图形界面
在命令行输入:
sudo service lightdm start然后按Ctrl + Alt + F7 即可恢复到图形界面。至此ubuntu安装Nvidia显卡驱动完成