tensorflow ) 초간단 GPU 사용하기

conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0

이제 window에서는 conda를 통해서 gpu를 바로 사용할 수 있게 되었다...Tensorflow공식문서

pip를 사용하여 TensorFlow 설치

wsl과 docker를 통해서 gpu를 사용한 학습(이것도 공식문서와 ms문서에 나온 것을 참고해서 함)을 진행해봤지만 tensorflow버전문제나 패키지간의 호환문제가 생겨 시간만 날리며 공식문서를 돌아다니다보니 window에서는 이 패키지설치만 하면 바로 사용이 가능하군요...(리눅스는 환경변수 추가 및 자동화 해줘야 함)

 

print(tf.config.list_physical_devices('GPU'))

위의 명령어로 사용 가능한 GPU 장치 목록을 확인할 수 있습니다.

 

p.s - 공식문서는 역시 영어 공식문서에서 찾아야 최신 업데이트 사항들이 적혀있네요...