橦言无忌

一个不想改变世界的程序媛

Nvidia GTX3090配置ubuntu20.04环境

前言

主机环境:DELL T7920图形工作站,2银牌4214R,24核 2.4G 64G内存,21TB M.2固态,RTX3090,ubuntu2004系统,主要用于深度学习视觉方向的开发环境搭建。

1,系统安装

ubuntu2004安装

2,显卡驱动

2.1 文件下载

下载链接

2.2 禁用nouveau

报错信息:

1
2
3
ERROR: The Nouveau kernel driver is currently in use by your system. This  driver is incompatible with the NVIDIA driver, and must be disabled before proceeding.
Please consult the NVIDIA driver README and your Linux distribution's documentation
for details on how to correctly disable the Nouveau kernel driver.

如果出现上述报错,则需要禁用nouveau:
1
2
3
4
5
6
7
8
9
10
# 首先打开或新建文件  
sudo vim /etc/modprobe.d/blacklist-nouveau.conf
# 在文件中添加如下内容
blacklist nouveau
options nouveau modeset=0
# 执行命令使其生效并重启
sudo update-initramfs -u
sudo reboot
# 查看禁用情况
lsmod | grep nouveau

2.3 驱动安装

显卡驱动:NVIDIA-Linux-x86_64-455.23.04.run

1
sudo bash NVIDIA-Linux-...

安装完成后用nvidia-smi可以查看显卡信息

3,基本环境

1
2
3
4
5
6
py37或py38
cuda11.0
cudnn8.0.4
tf2.5(tf-nightly)或 tf1.15.4
pytorch1.7
keras2.3
// 代码折叠