본문 바로가기
AI/Tensorflow

텐소플로우를 윈도우에 설치하기

by 신군. 2018. 12. 1.
반응형




머신러닝을 배우기 위해 텐소플로우를 설치하고 tutorial을 다 따라하려고 합니다.

윈도우에 설치해봅니다.


텐소플로우는 윈도우를 처음부터 지원하지 않습니다. 그래서 Docker라는 툴을 이용해서 호스트 컴퓨터와 클라이언트 컴퓨터를 유연하게 컨테이너 형식으로 유연하게 연결 시켜 사용할 수 있게 만듭니다.


1. docker toolbox 설치


https://www.docker.com/products/docker-toolbox


- NDIS5와 NDIS6는 호스트 네트워크 필터 드라이버 입니다.

- NDIS6는 비스타나 그 이후부터, NDIS5는 하위버전에서 자동으로 선택됩니다.

- 강제로 networktype을 NDIS5로 설정하면 네트워크 필터가 바뀌게 됩니다.

- 자세한건 https://en.wikipedia.org/wiki/Network_Driver_Interface_Specification

-> 구지 그럴필요없으니 체크 x

설치합니다.


2. Docker Quickstart Teminal을 사용해서 second docker machine vdocker 만들기


$ docker-machine ls



$ docker-machine create vdocker -d virtualbox



$ docker-machine ls



이제 windows에서 cmd prompt를 키고 명령어를 칩니다.

FOR /f "tokens=*" %i IN ('docker-machine env --shell cmd vdocker') DO %i




docker run -it -p 8888:8888 gcr.io/tensorflow/tensorflow


http://stackoverflow.com/questions/35582875/unable-to-start-tensorflow-within-docker-on-windows

위의 명령어를 치면 stuck됩니다. 그래서 아래와 같이 또 쳐주면 됩니다.

docker run -it -p 8888:8888 gcr.io/tensorflow/tensorflow:latest-devel

에제 제대로 실행되니 python으로 tensorflow를 실행해봅니다.


$ python
...
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print sess.run(hello)
Hello, TensorFlow!
>>> a = tf.constant(10)
>>> b = tf.constant(32)
>>> print sess.run(a + b)
42
>>>



설치를 완료했습니다~


다시 킬때는 3가지를 해야합니다.

1. docker-machine start vdocker

2. cmd에서 

FOR /f "tokens=*" %i IN ('docker-machine env --shell cmd vdocker') DO %i

3. cmd에서 

docker run -it -p 8888:8888 gcr.io/tensorflow/tensorflow:latest-devel

하면 껏다 켜도 재실행됩니다.


reference


http://www.netinstructions.com/how-to-install-and-run-tensorflow-on-a-windows-pc/

http://purip.blogspot.kr/2016/03/tensorflow.html



출처: http://whatx4.tistory.com/69 [whatx4]

반응형

'AI > Tensorflow' 카테고리의 다른 글

CUDA_ERROR_NO_BINARY_FOR_GPU  (0) 2023.01.12