본문 바로가기

트렌드 한눈에 보기/학계 트렌드

[2023 ver.] tensorflow에서 GPU를 사용하는 방법 Windows

개발 환경은 하루가 멀다 하고 바뀌기 때문에, 종이 책으로 공부하는 것은 절대로 추천하지 못한다. 심지어 인터넷 자료들조차도 라이브러리 업데이트 속도를 따라가지 못하는 경우가 허다하다. 아래 자료가 가장 최신 (2023년 11월 기준)으로, Windows 11 / Tensorflow 2.10 이하 / python 3.9 이하에서 GPU를 사용하는 방법을 정리한 것이다.

pip로 TensorFlow 설치

 

pip로 TensorFlow 설치

Learn the latest in machine learning, generative AI, and more at WiML Symposium 2023Register 이 페이지는 Cloud Translation API를 통해 번역되었습니다. Switch to English pip로 TensorFlow 설치 컬렉션을 사용해 정리하기 내 환경

www.tensorflow.org

 

"엥 뭐야 내 개발 환경이랑 다르네~ 나는 Tensorflow 2.14란 말야" 하는 사람도 있겠지만 노노. GPU를 사용하기 위해서는 Tensorflow 2.10 이하로 사용해야 한다. 예전에는 (예전이라고 해봤자 불과 1~2년 전이다!) tensorflow-gpu라는 모듈이 있었는데 현재는 tensorflow로 통합된 상황이다. 이러니 인터넷에서 자료를 찾아도 헷갈린다.

 

여튼 위 자료를 충실히 따라가기만 하면 GPU를 활용하면서 Tensorflow를 사용할 수 있다.