Ubuntu20.04 + RTX 3090(兼容RTX 2080 Ti) + Pytorch1.7配置方法

背景介绍：由于在Ubuntu16.04系统上安装RTX 3090显卡驱动有点吃力(各种Error和不兼容)，使用最新Ubuntu20.04系统搭配最新的RTX 3090显卡配置最新的Pytorch【(*^▽^*)】

前期准备：

              1、Ubuntu20.04下载：Ubuntu20.4_amd64_desktop.iso 

              2、UNtebootin光盘刻录软件下载：unetbootin，选择Windows下载

              3、NVIDIA-455显卡驱动安装：NVIDIA-Linux-x86_64-455.45.01，下载455最新系列

              4、Anconda下载：Anaconda3-2020.11-Linux-x86_64.sh

系统安装：

             1、选择最小系统安装，选择其它选项系统安装类型

             2、分区设置重点：需要设置保留BIOS分区，启动引导(boot loader)选项为/biosgrub所在分区作为加载项(否则可能会出现无法进入系统的情况)

/	主分区	ext4	10G
/boot	逻辑分区	ext4	1G
/swap	逻辑分区		10G
/efi	逻辑分区	ext4	10G
/home	逻辑分区	ext4	剩余
/biosgrub <boot manager>	保留bios分区		10G

显卡安装：

             1、禁用nouveau

#卸载旧版本nvidia*
 sudo apt-get purge nvidia*
 
#查看nouveau,无变化则为已经禁用
lsmod | grep nouveau
 
#禁用自带nouveau驱动
sudo gedit /etc/modprobe.d/blacklist.conf
 
#加入以下设置
blacklist nouveau
options nouveau modeset=0
 
#刷新设置
sudo update-initramfs -u
 
#重启并查看
reboot
 
#无变化则禁用成功
lsmod | grep nouveau

             2、安装显卡驱动NVIDIA-455

sudo ./NVIDIA-Linux-x86_64-390.42.run -no-x-check -no-nouveau-check -no-opengl-files

     可能出现二次进入系统时，系统无法进入图形界面：

# 解决办法
sudo ubuntu-drivers autoinstall
sudo systemctl set-default graphical.target
sudo service lightdm start

查看一下显卡驱动： 

 

Pytorch1.7安装：

conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch

测试一下Pytorch： 

最后的最后，别忘了设置系统永不待机(设置--电源--电池--永不锁屏) ！