[Cuda] GTX 3090 사용 시 tensorflow graph 생성이 느린 문제

그래픽 카드 3090에서 tensorflow를 사용할 때 모델을 생성하는 부분이 상당히 느립니다

 

3090을 사용하기 위해서는 cuda 11.0 이상을 사용해야하는데 그에 최적화된 tensorflow version이 따로 있습니다

 

일반적으로 conda install tensorflow-gpu를 사용하면 2.2 version으로 설치가 되고 cudatoolkit도 10.0으로 깔립니다

 

이 상태에서 tensorflow를 돌릴 때 처음 모델을 생성하는게 매우 느려지는데

 

이 문제를 해결하기 위해서는

 

$ conda install cudatoolkit          # cudatoolkit이 11.0 버전으로 깔림

                                                                     # cuda가 다른 버전으로 깔리면 cudatoolkit=11.0 으로 지정해주시면 됩니다

$ pip install tensorflow-gpu        # tensorflow-gpu가 2.4 버전으로 깔림

 

이후에 tensorflow를 사용하면 제대로 작동이 되시는걸 보실 수 있습니다