RTX 3090에 맞는 Tensorflow 설치(Anaconda이용)

최근에 대학원에서 RTX3090을 새로 설치하면서 이를 활용해보고자 했는데 Tensorflow 버전 때문에 잘 작동하 않았다. 조금 알아보지 Tensorflow 최신버전에서만 잘 작동하는 것 같다.

 

RTX3090을 쓰기 위한 몇가지 버전 관련한 요구사항은 아래와 같다.

• Tensorflow 2.5 이상
• cudatoolkit 11.2 이상
• cudnn 8.1이상

다음 링크에서 확인할 수 있다: https://www.tensorflow.org/install/source#tested_build_configurations

이를 설치하기 위하여 아래 커맨드를 실행해 주었다. cudatoolkit, cudnn 버전을 특정해주지 않았는데, 2021년 9월 30일 기준으로 cudatoolkit은 11.3.1, cudnn은 8.2.1 버전이 설치된다. Tensorflow의 경우는 pip로 설치하는게 깔끔하게 설치되었다.

 

conda create -n tf_3090 python==3.8 numpy==1.19.1
conda install cudatoolkit cudnn
pip install tensorflow-gpu==2.5.0

이렇게 설치하고 나니 RTX 3090을 원활하게 쓸 수 있었고, 성능을 측정해보니 기존 Titan RTX에 비해 30%는 빨라지는 것 같다.