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

ubuntu20.04安装pytorch深度学习环境(GPU RTX3060)_第1张图片

 下面是我的显卡型号

ubuntu20.04安装pytorch深度学习环境(GPU RTX3060)_第2张图片

2.下载驱动

去英伟达官网下载对应版本的显卡驱动。

驱动下载地址

ubuntu20.04安装pytorch深度学习环境(GPU RTX3060)_第3张图片

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

ubuntu20.04安装pytorch深度学习环境(GPU RTX3060)_第4张图片

 7. 根据提示完成操作(不同电脑提示略有不同,按照提示一步步操作即可)

(1)许可信息,选择左侧Accept(接受)

ubuntu20.04安装pytorch深度学习环境(GPU RTX3060)_第5张图片

(2)提示有冲突: The distribution-provided pre-install script failed! Are you sure you 

want to continue?  这个信息不必理会,选择左侧Continue installation

(3)开启构建NVIDIA核心模块。

(4)提示32位兼容库?我选侧的yes.

(5) 提示信息,回车

ubuntu20.04安装pytorch深度学习环境(GPU RTX3060)_第6张图片

 (6)启动X服务运行NVIDIA配置文件,自动更新X配置文件,选择左侧yes。

ubuntu20.04安装pytorch深度学习环境(GPU RTX3060)_第7张图片

(7) 提示信息,安装成功.

 8. 验证是否安装成功

 9. 恢复图形界面

 在命令行输入:

sudo service lightdm start

然后按Ctrl + Alt + F7 即可恢复到图形界面。至此ubuntu安装Nvidia显卡驱动完成

你可能感兴趣的:(Python,深度学习,pytorch,深度学习,人工智能)