ubuntu20.04LTS에 Tensorflow 설치하기

(1) 그래픽 드라이버 설치 

# 각자 버전에 맞는 그래픽 드라이버 설치
# http://www.nvidia.com/Download/Find.aspx?lang=en-us
sudo apt install nvidia-driver-460

- 리눅스에서 그래픽드라이버 설치 확인하는 방법 -

# 설치확인
cat /proc/driver/nvidia/version

# 버전체크
apt-cache search nvidia

# GPU 확인
lspci -k
# or GPU, driver version 확인
nvidia-smi

(2) Tensorflow, cuda, cudnn 호환 가능한 버전 확인하기

https://www.tensorflow.org/install/source_windows?hl=ko#tested_build_configurations

※ Jacinto mmdetection 프로그램에 설치를 위한 셋팅이라 cuda 10.2, cudnn 10.2로 설치를 진행했다.

(3) CUDA 설치

# https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
sudo sh cuda_10.2.89_440.33.01_linux.run
- Continue
- accept
- driver (x) 표시 해제

sudo vi ~/.bashrc # 하단에 추가
===========================
export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda- 10.2/lib64\${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
===========================


# 설치확인
nvcc -V

(4) cuDNN 설치

# 설치버전 확인
wget https://developer.nvidia.com/compute/machine-learning/cudnn/secure/8.2.0.53/10.2_04222021/cudnn-10.2-linux-x64-v8.2.0.53.tgz

# cudnn.tar.gz
tar xvf cudnn-10.2-linux-x64-v8.0.5.39.tgz

# include, lib64 폴더에 있는 파일 cuda-10.2 폴더에 덮어쓰기
sudo cp ./cuda/include/* /usr/local/cuda-10.2/include
sudo cp -P ./cuda/lib64/* /usr/local/cuda-10.2/lib64
sudo chmod a+r /usr/local/cuda-10.2/lib64/libcudnn*

#cudnn 버전확인(8버전)
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
#7버전
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
# 재시작
sudo reboot

(5) Tensorflow-gpu 설치

# Tensorflow-gpu 설치 
pip install tensorflow-gpu==2.4.0

- gpu 가동 유무 확인 -

# Tensorflow - GPU 설치 확인

(1)
import tensorflow as tf

print(tf.__version__)
# 1.14.0

tf.test.is_gpu_available(
    cuda_only=False,
    min_cuda_compute_capability=None
)
# True

(2)
from tensorflow.python.client import device_lib
device_lib.list_local_devices()