install tf-gpu 2.4.0

ubuntu 18.04

 

sudo apt-get update

sudo apt-get upgrade -y

 

Install Nvidia-driver 460

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

apt-cache search nvidia | grep nvidia-driver

sudo apt-get install nvidia-driver-455

or

sudo ubuntu-drivers autoinstall

reboot

# Check installed nvidia-driver version

nvidia-smi

Install cuda 11.0 for tensorflow 2.4.0

wget http://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda_11.0.2_450.51.05_linux.run

sudo sh cuda_11.0.2_450.51.05_linux.run

# Check installed cuda version

nvcc -V

 

Install cudnn 8.0 for tensorflow 2.4.0

wget developer.nvidia.com/compute/machine-learning/cudnn/secure/8.0.5/11.0_20201106/Ubuntu18_04-x64/libcudnn8_8.0.5.39-1+cuda11.0_amd64.deb

wget developer.nvidia.com/compute/machine-learning/cudnn/secure/8.0.5/11.0_20201106/Ubuntu18_04-x64/libcudnn8-dev_8.0.5.39-1+cuda11.0_amd64.deb

 

### Nvidia driver 삭제

sudo apt-get -y remove --purge nvidia*

sudo apt-get -y autoremove

sudo apt-get -y autoclean

 

### CUDA 삭제

sudo apt-get -y remove --purge cuda*

sudo rm -rf /usr/local/cuda*

 

### CuDNN 삭제

sudo apt-get -y remove --purge cudnn*

 

Install tensorflow-gpu 2.4.0 on python virtual environment.

Refer from

1. www.tensorflow.org/install/pip?hl=ko

2. whereisend.tistory.com/30

1) install python virtual environment

sudo apt update

# install python virtual environment
sudo apt install python3-dev python3-pip python3-venv

# create virtual environment

python3 -m venv --system-site-packages ./venv

# activate virtual env.

source ./venv/bin/activate  # sh, bash, or zsh

# update virtual env.

pip install --upgrade pip

# deactivate

deactivate

 

2) install tensorflow-gpu==2.4.0

(venv) pip install tensorflow-gpu==2.4.0

# (1) Check tensorflow installation

(venv) python -c "import tensorflow as tf;print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

# (2) Check installed tensorflow version

(venv) python

(venv#) import tensorflow as tf

(venv#) tf.__version__

# (3) Check tensorflow - GPU usability

from tensorflow.python.client import device_lib

device_lib.list_local_devices()