본문 바로가기

카테고리 없음

window anaconda 환경에서 Tensorflow GPU 사용하기

피똥을 여러번 싸고나서야 그 방법을 파악하고 업로드 한다.

 

일단 이걸 제일 먼저 봐야한다.

그러니까 GPU 쓸거면 pip install tensorflow 하면 안된다. 지금 시간 기준으로 2.11 버전이 깔리고, 이녀석은 윈도우상에서의 gpu 지원이 안된다.

 

그러면 뭘 해야하냐?

요 밑에 것들을 설치해줘야 한다.

 

예를들어 2.10 버전을 쓰고 싶으면 파이선 3.7을 설치하고, 쿠다는 11.2를 설치하자. 그리고 cudnn은 8,1로 깔자.

 

쿠다와 cudnn의 경우 bin 폴더를 시스템 PATH에 등록만 해주면 된다.

 

이걸 다 잘하고 나서

import tensorflow as tf
from tensorflow.python.client import device_lib

print(device_lib.list_local_devices())
 

위 코드를 돌리면

이런 코드를 볼 수 있다.

 

이상 정리 끝.