TF
-
Tensorflow에서 사용하는 데이터 배열 Tensor 사용하기(constant, Variable)Data Science/Tensorflow 2022. 10. 24. 22:12
Numpy에서 Numpy.array를 사용하듯이 Tensorflow에서는 tensor를 사용한다. 본 포스팅에서는 Tensor란 무엇이고, Tensor를 어떻게 다룰수 있는지 알아본다. 1. Tensorflow.constant contant tensor는 생성후 값의 수정이 불가능한 Tensor이다. 1.1 Constant tensor 생성 tensor = tf.constant([1,2,3,4,5]) 생성되는 tensor의 데이터 타입을 지정해줄 수도 있다. tensor = tf.constant([1,2,3], dtype='float32') 1.2 Constant tensor 정보 확인 생성된 tensor로부터 아래의 정보를 얻을수 있다. tensor = tf.constant([1,2,3,4,5]) pr..
-
Tensorflow 세팅 확인(버전확인, GPU확인, GPU 지정, 지정된 GPU 확인)Data Science/Tensorflow 2022. 10. 24. 21:06
본글은 Tensorflow 설치 및 Cuda 설치를 다루지 않습니다. 두가지 세팅 완료 후 확인 부분을 다루고 있습니다. 1. 라이브러리 Import import tensorflow as tf 2. Tensorflow 버전확인 print(tf.__version__) 3. GPU 상태 확인 "!"를 통해서 터미널에서 사용가능한 명령어를 사용할 수 있다. 터미널(또는 cmd)에서 nvidia-smi 명령어를 통해서 gpu의 상태를 확인할 수 있다. !nvidia-smi 위 그림에서 확인 가능한 GPU정보는 다음과 같다. 현재 1개의 GPU가 0번으로 연결되어 있으며, 해당 GPU는 RTX A5000이다. (만약 여러개의 GPU가 연결되어 있다면 순차적으로 0~n번의 번호를 부여받는다.) 4. 사용할 GPU ..