더북(TheBook)

4. 텐서플로 GPU 버전을 설치

아나콘다 프롬프트를 관리자 권한으로 실행해서(또는 주피터 노트북에서 New 버튼 > Terminal을 열어서) 다음과 같이 tensorflow gpu 버전을 설치한다.

>>> pip install tensorflow-gpu

또는 다음처럼 특정 버전을 지정해 설치할 수도 있다.

>>> pip install tensorflow-gpu==2.0

이후 주피터 노트북에서 다음처럼 실행하면 정상적으로 설치된 것을 확인할 수 있다.

>>> import tensorflow as tf
>>> print(tf.test.is_gpu_available())
True
신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.