Tensorflow 1.x 버전 설치하기. feat conda

Conda 가상환경 만들기는 아래 링크 참고

https://barrer.tistory.com/97

아나콘다 가상환경 만들기 conda create

 

문제 상황

tensorflow 1.X 버전대로 만들어진 코드를 2.X 버전에서 돌리면 무수히 많은 오류 코드를 만나게 된다.

그때 각각의 모듈들을 변경하거나 2.X 버전에서 1.X 버전이 돌아가게끔 다음과 같은 방식을 쓸 수 있다.

 

import tensorflow.compat.v1 as tf

tf.disable_v2_behavior()

이렇게 하더라도 자잘한 코드 오류를 일일이 하나씩 대응하는게 너무 귀찮기도 하고 보통은 pytorch를 많이 쓰다보니

빠르게 tensorflow 데모만 확인하고자 가상환경에서 돌리기로 결정했다.

가상환경 생성

우선은 conda가 설치된 환경에서 tensorflow 가상환경을 생성한다.

tensorflow 1.x 버전 설치를 위해 python은 3.7 버전을 이용한다.

conda create -n tensorflow python=3.7

conda activate tensorflow

그 다음 Tensorflow, Cudatoolkit, cudnn을 설치한다.

conda install tensorflow==1.15.0

conda install -n tensorflow cudatoolkit=10.0

conda install -n tensorflow cudnn=7.6.0

tensorflow 1.15.0 버전을 설치하였고 이에 맞는 cuda, cudnn을 위와 같은 버전으로 설치하였다.

Tensorflow 설치 확인

설치가 완료되면 GPU 사용도 잘 되는지 확인해본다.

GPU 잘 잡히는지 확인

python

import tensorflow as tf

print("GPU Available: ", tf.test.is_gpu_available())

현재 GPU로 사용하는 NVIDIA GeForce RTX 3090이 제대로 확인 되는 것을 볼 수 있다.