본문 바로가기
공부하는 중~~/분류하기 애매한 것들

[Ubuntu 20.04] RTX3090 딥러닝 환경세팅 공유

by 임대추 2020. 12. 3.

OS : ubuntu 20.04

GPU : RTX3090 X 2EA

CUDA : 11.1

CUDNN : 8.0.3

Tensorflow : tf.nightly (2.3.0으로 진행할시 모델마다 다르지만 일부 모델은 작동하지 않음)

Keras : 2.4.3 (최신 버전)

위와 같은 세팅으로 멀티 gpu까지 정상적으로 동작합니다.

반응형

댓글