본문 바로가기
AI/텐서플로우(구글)

텐서플로우 설치

by N2info 2017. 5. 1.

0. 필요사항

  

텐서플로우 파이썬 API는 파이썬 2.7과 파이썬 3.3+을 지원함.

파이썬 3.5.2로 설명함


텐서플로우 GUI버전은 Cuda Toolkit 7.5 과 cuDNN v4 와 가장 잘 작동함.


1. pip 설치

# Python 2
sudo apt-get install python-pip python-dev

# Python 3
sudo apt-get install python3-pip python-dev




2. 적절한 텐서플로우 바이너리를 선택


# Ubuntu/Linux 64-bit, GPU 버전, Python 3.5
# CUDA toolkit 7.5 와 CuDNN v4 필수. 다른 버전을 사용하려면 아래 "소스에서 설치" 섹션을 참고하세요.
$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.9.0-cp35-cp35m-linux_x86_64.whl



3. 텐서플로우 설치


# Python 2
$ sudo pip install --upgrade $TF_BINARY_URL

# Python 3
$ sudo pip3 install --upgrade $TF_BINARY_URL



인텔 GPU이다. 여기서 스톱 ㅠㅠ


4. 텐서플로우 설치테스트  ...  아직



GPU 확인법 

lspci | grep VGA





참조사이트 : https://tensorflowkorea.gitbooks.io/tensorflow-kr/content/g3doc/get_started/os_setup.html