Tensorflow2 hello

개요

머신러닝, 딥러닝 강의를 검색하면 이론부터 여러개를 보아야 해서 개발자 입장에서는 손이 바로 안가게 됩니다. 그래서 일단 hello를 출력하고 나면 이론 동영상들을 보는데 좀 더 몰입을 하게 되지 않을까 해서 hello를 출력 해보았습니다.

 

파이썬 버젼 이슈

python 3.7.9버젼에서 설치가 가능합니다. 3.9로 했더니 안되더라구요.

 

Tensorflow 설치

pip install --index-url https://pypi.org/simple tensorflow --trusted-host pypi.org --trusted-host files.pythonhosted.org

설치는 위 커맨드로 설치 하면 됩니다. 설치 하면 2021년 3월 기준으로 최신 버젼인 2.4.x버젼이 설치 됩니다. 더 낮은 버젼을 설치 하려면 버젼을 특정해서 설치 해야 합니다. 2.x버젼을 Tensorflow2라고 합니다. 텐서플로2부터는 사용방법이 티가 나게 바뀌었습니다.

 

import tensorflow as tf

hello = tf.constant('hello')
print(hello)

tf동영상을 보면 2017년부터 2020년까지 만들어진 비디오에서 session을 생성해서 .run()하게 되어있는 부분이 있는데요 이 부분은 Tensorflow2로 올라오면서 텐서를 호출하면 실행되게끔 바뀌었습니다.

 

결과

tf_xla_enable_xla_devices not set
tf.Tensor(b'hello', shape=(), dtype=string)

 

Gpu사용하기

Nvidia Gpu가 있다면 Cuda를 받아서 쓰면 빠릅니다.

developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64

CUDA Toolkit 11.2 Update 1 Downloads

위 사이트에서 받아서 설치하시면 됩니다. 2.8기가 입니다.