본문 바로가기

Industry 4.0/Install

TensorFlow 2.0 설치

728x90
반응형

conda create -n tensorflow2 python=3.7

 

Proceed ([y]/n)? y

설치완료

 

제대로 설치되었는지 확인하기

activate tensorflow2

 

좌측에 base 에서 tensorflow2 로 바뀐것을 확인 할 수 있다

이제 이 상태에서 

pip install tensorflow-gpu 

를 입력해준다

 

다운로드 완료

 

GPU가 정상 작동하는지 확인해보기 위해 아래의 코드를 입력해본다

 

python -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

 

(1000, 1000) 크기의 정규분포 랜덤값의 총합이 반환되는 것을 확인하였다

설치 완료!

 

 

 

 

728x90
반응형