Ubuntu20.04 - CUDA, tensorflow 설치하기

텐서플로우를 설치하는데 공식 문서에는 18.04 를 기준으로 안내를 하고 있다.

Ubuntu20.04 를 기준으로 설치하는데 몇번을 삽질하고 쉽게 설치하는 방법을 찾아 정리 해본다.

아래의 명령어를 통해 GPU 드라이버와, 기초 라이브러리를 설치한다.

 

# Add NVIDIA package repositories
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub
sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"
sudo apt-get update

# Install CUDA , libcudnn8 
sudo apt-get -y install cuda libcudnn8

패키지 매니저로 설치하면 간편하게 GPU 드라이버와 필수 패키지들이 모두 설치가 된다.

재부팅을 하고 nvidia-smi 명령어를 통해 GPU 가 잘 인식 되는지, CUDA 버전과 함께 확인해본다.

Tesla T4 가 인식 되었으며 CUDA 11.4 가 설치된 상태

 

---

tensorflow 설치하기 

패키지 의존성, 충돌등 여러 문제를 겪고 나서부터는 가상 환경을 적극적으로 사용하고 있다.

아래는 가상환경을 설정하는 명령어이다. 

시스템에 설치할 경우 이 단계는 무시해도 된다.

# 홈 폴더로 이동
cd ~

# 가상 환경 설정
python3 -m venv --system-site-packages ./venv

# 가상 환경 실행
source ./venv/bin/activate

 

참고: https://www.tensorflow.org/install/pip#virtual-environment-install

pip를 사용하여 TensorFlow 설치

 

# python tensorflow 설치 

pip install --upgrade tensorflow

 

이후 텐서플로 코드를 실행하고 

nvidia-smi 에서 실행한 프로세스가 뜨는걸 확인해보면 된다.

 

우분투 18.04 에서 20.04 로 20.04 에서 18.04로.. cudnn 라이브러리 때문에 몇번을 지웠다 다시 깔고 삽질했는지 모르겠다 ㅠ

아래는 겪었던 여러 문제상황들..

cudnn 라이브러리 인식이 안된 상황

라이브러리 충돌로 gpu 인식이 안되는 상황

cudnn 라이브러리 버전 때문에 프로그램 코드가 실행이 안되는 상황

텐서플로 설치후 detectron2 설치가 안되는 상황

 

정상적일때 상황