ubuntu18.04 RTX3080 搭建 tensorflow 环境

ubuntu18.04  RTX3080  搭建 tensorflow 环境

RTX3080安装tensorflow v2.5dev版本

1、安装nvidia驱动

禁止自带 nouveau nvidia 驱动

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

添加：

blacklist nouveau

options nouveau modeset=0 

保存退出， 更新配置文件

sudo update-initramfs -u

添加源

sudo add-apt-repository ppa:graphics-drivers/ppa

sudo apt-get update

sudo apt-get install nvidia-driver-456

2、安装 CUDA

在nvidia官网直接下载cuda11

sudo chmod a+x cuda_11.1.0_455.23.05_linux.run

sudo ./cuda_11.1.0_455.23.05_linux.run

export PATH=/usr/local/cuda-11.1/bin${PATH:+:${PATH}}

export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

3、配cudnn

在nvidia官网直接下载8.0

tar -zxvf cudnn-11.1-linux-x64-v8.0.5.39.tgz

sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda-11.1/lib64/

sudo cp cuda/include/cudnn.h /usr