tensorflow gpu 설치 방법 (window10)

목차

• 가상 환경 만들기
• anaconda로 tensorflow gpu 설치하기

 

개발 환경

- window10

- anaconda python 3.7 64bit

 

아나콘다를 이용해 tensorflow-gpu를 설치하겠습니다. 가상 환경 만들기는 건너뛰어도 상관없으나 설치 오류가 생긴다면 새로운 가상 환경을 만들어서 설치해주세요.

 

1. 가상환경 만들기

conda create -n 'virtual_name' python=3.7 anaconda

 

생성된 가상환경 확인

conda env list

# conda environments:
#
base                  *  C:\Anaconda3
tensorflowgpu            C:\Anaconda3\envs\tensorflowgpu

 

가상 환경에 진입

conda activate 'virtual_name'

가상 환경에 진입하면 맨 왼쪽에 가상 환경 이름이 나옵니다.

 

2. tensorflow gpu 설치하기

anaconda prompt를 관리자 권한으로 실행시켜 줍니다. 자신의 가상 환경에 진입한 후 conda 명령어를 실행시켜줍니다.

 

conda install -c anaconda tensorflow-gpu

명령어를 실행시켜 준 뒤 빨간색 박스 친 곳이 있는지 확인합니다.

y를 눌러 계속 진행하면 설치가 완료됩니다.

 

- 설치확인 방법

명령 프롬프트 창에 

(tensorflowgpu) C:\WINDOWS\system32>python

Python 3.7.10 (default, Feb 26 2021, 13:06:18) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.

 

>>> import tensorflow

2021-06-21 21:22:13.828056: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_101.dll

 

>>> print(tensorflow.__version__)

2.1.0

이렇게 나오면 설치가 완료됩니다.

 

 

- 더 알아보기

https://anaconda.org/search?q=tensorflow-gpu 

Search :: Anaconda.org

 

anaconda 사이트에 더 다양한 패키지를 보고 싶으면 위 사이트에서 찾아본다.