[tensorflow 설정] gpu 세팅하기

pytorch로 이미지 분석을 하려다 보니 생각보다 자료가 많이 없더라구요.

 

그래서 먼저 tensorflow로 모델을 제작하고 그후에 pytorch로 진행을 하고자 합니다. pytorch는 생각보다 쉽게 세팅이 가능하더라구요.

 

학교에서는 GTX1050ti, 집에서는 GTX1060 으로 진행하였습니다.

 

세팅의 경우 똑같습니다. 위 두개의 GPU에서 CUDA는 11.0을 이용하였고, cuDNN의 경우 8.0.5를 이용하였습니다.

 

그리고 pip install을 진행할 때 다음과 같이 진행해줍니다.

 

pip install tensorflow==2.4.0
pip install tensorflow-gpu==2.4.0

 

그리고 나서 python을 틀고 tensorflow를 import시키면

 

 

성공했다는 무엇인가가 나오고 gpu를 확인하기 위해 다음을 추가로 쳐줍니다.

 

from tensorflow.python.client import device_lib
device_lib.list_local_devices()

 

그리고 무엇인가가 주르륵 나오고 

 

GPU가 있음을 확인할 수가 있죠.

 

다음 포스팅은 gpu로 돌리는 방법에 대해서 추가적으로 올리도록 하겠습니다.