[Ubuntu 18.04] Tensorflow==2.5.0 설치하기

ubuntu 18.04의 anaconda 환경에서 tensorflow 설치하는 법을 알아보자..!

생각보다 매우매우 간단하다. 

---

우선, 본인 컴퓨터에 설치되어있는 패키지들의 버전을 확인하자.

 

1. 가용가능한 GPU 확인

lspci | grep -i VGA

 

2. Python version 확인

만약 conda 환경에서 확인한다면, conda activate을 해준 후 파이썬 버전을 확인해준다.

python --version

3. 설치된 Cuda version 확인

nvcc -V

4. Tensorflow version 찾기

https://www.tensorflow.org/install/source#tested_build_configurations

소스에서 빌드  |  TensorFlow

여기에서 내 python 버전과 cuda 버전에 맞는 tensorflow 버전을 찾는다.

5. Tensorflow 설치

------- CPU 버전

pip install tensorflow==2.5.0

------- GPU 버전

pip install tensorflow-gpu==2.5.0

6. 설치 확인

이렇게 하면 촤라락~! 설치가 되고, tensorflow 설치를 확인하면 설치가 되었음을 볼 수 있다.