tensorflow docker에서 설치하기

2018. 4. 28. 16:21Python-이론/python-인공지능2

tesnsorflow docker에서 설치하기



도커에서 텐서플로우를 실행시키려면 docker run -it -p 8888:8888 gcr.io/tensorflow /tensorflow:latest-devel을 적어 주면 된다. 그럼 ubuntu 안으로 들어가진다. 필자의 경우에는 python2에서는 바로 작동이 되었지만 python3에서는 이상하게도 작동이 제대로 되지 않았다. 그래서 삽질을 거듭한 끝에 방법을 알았으니 우선 apt-get install python3-pip 쳐서 다운 후 pip3 install tensorflow을 쳐서 사용하면 된다. 이상한 책 때문에 삽질함 ㅂㄷㅂㄷ 그리고 설치 후 확인을 위해 python3를 쳐서 쉘에 들어간 후




#이거 친 후 오류나면 잘못 깔린 것이다......
>>import tensorflow as tf
# 상수 선언 
>>a = tf.constant(1234)
>>b = tf.constant(5000)
# add_op는 상수가 아니라 데이터 플로 객체를 만든것이여 
>>add_op = a+b
#session 실행
>>sess = tf.Session()
>>res = sess.run(add_op)
>>print(res)


오류 없이 실행된다면 제대로 설치 된것이 맞다.


참고 사이트