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 --uninstall

3. 禁用nouveau驱动

编辑blacklist.conf文件

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

在文本的最后添加:

blacklist nouveau
options nouveau modeset=0

然后更新内核配置

sudo update-initramfs -u

重启系统，执行下面命令 ，如果没有屏幕输出，说明nouveau禁用成功!

lsmod | grep nouveau

4. 进入到字符界面

使用 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 lightdm

6. 安装驱动

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显卡驱动完成