Nvidia GeForce RTX 3090에서 Tensorflow 1(1.15) 실행하기

OS : Ubuntu 20.04 LTS

GPU Device : RTX 3090 2대

 

연구를 하면서 Tensorflow 1버전을 사용한 딥러닝 모델 코드를 RTX 3090 환경에서 수행을 해야 하는 상황이 되었다.

그런데 RTX 3090가 인식하는 CUDA의 가장 낮은 버전은 11.1인데 Tensorflow1.x은 CUDA 버전이 10인 환경에서 수행이 가능하다.

 

본론부터 말하자면 NGC(Nvidia GPU Cloud)를 사용하여 문제를 해결했다.

아래는 내 도커 파일의 일부이다.

######### 1. NGC의 Tensorflow 공식 이미지
ARG BASE_IMAGE=nvcr.io/nvidia/tensorflow:20.10-tf1-py3
FROM $BASE_IMAGE

WORKDIR /workspace

# export timezone - for python3-tk install
ENV TZ=Asia/Seoul
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone \
# 필수 설치요소들 설치
######### 2. GPG Key 변경
 && apt-key del 7fa2af80 \
 && apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/3bf863cc.pub \
 && apt-get update \
 && apt-get install -y build-essential \
    python3 \
    python3-pip \
    python3-tk \
    python3-dev \
    gcc \
    openjdk-8-jdk \
    libx11-6 \
    python-dev \
    wget \
 .
 .
 .
 생략

 

 

참고: 

https://sseongju1.tistory.com/62