본문 바로가기
Python

Tensorflow 2.0 release 이후 설치 방법

by bryan.oh 2019. 10. 21.
반응형

Install TensorFlow with pip

 

요약하자면 

pip install tensorflow 와 같이 뒤에 버전을 안치면 2.0 으로 설치가됩니다.

아직 샘플 코드가 1.x 버전으로 나온게 많기 때문에 1.x 버전을 설치하려면

pip install tensorflow==1.15rc2  와 같이 뒤에 버전을 명시하면 됩니다.

cpu와 gpu 버전을 모두 포함한 1.15rc2 버전이 좋겠네요.

cpu버전만 할거면 pip install tensorflow==1.14

gpu버전만 할거면 pip install tensorflow-gpu==1.14

 

 

TensorFlow 2 packages are available

  • tensorflow —Latest stable release for CPU-only
  • tensorflow-gpu —Latest stable release with GPU support (Ubuntu and Windows)
  • tf-nightly —Preview build (unstable). Ubuntu and Windows include GPU support.

Older versions of TensorFlow

For the 1.15 release, CPU and GPU support are included in a single package:

  • tensorflow==1.15rc2 —1.x release. Ubuntu and Windows include GPU support

For releases 1.14 and older, CPU and GPU packages are separate:

  • tensorflow==1.14 —Release for CPU-only
  • tensorflow-gpu==1.14 —Release with GPU support (Ubuntu and Windows)

System requirements

  • pip 19.0 or later
  • Ubuntu 16.04 or later (64-bit)
  • macOS 10.12.6 (Sierra) or later (64-bit) (no GPU support)
  • Windows 7 or later (64-bit) (Python 3 only)
  • Raspbian 9.0 or later

 

 

728x90
반응형

댓글