Google is committed to advancing racial equity for Black communities. See how.

ติดตั้ง TensorFlow ด้วย pip

มีแพ็คเกจ TensorFlow 2

  • tensorflow - รุ่นเสถียรล่าสุดพร้อมรองรับ CPU และ GPU (Ubuntu และ Windows)
  • tf-nightly สร้างตัวอย่าง (ไม่เสถียร) Ubuntu และ Windows รวมถึง การรองรับ GPU

TensorFlow เวอร์ชันเก่ากว่า

สำหรับ TensorFlow 1.x แพ็กเกจ CPU และ GPU จะแยกจากกัน:

  • tensorflow==1.15 ปล่อยสำหรับ CPU เท่านั้น
  • tensorflow-gpu==1.15 ปล่อยด้วย การสนับสนุน GPU (Ubuntu และ Windows)

ความต้องการของระบบ

  • Python 3.6–3.9
    • การสนับสนุน Python 3.9 ต้องการ TensorFlow 2.5 หรือใหม่กว่า
    • การสนับสนุน Python 3.8 ต้องการ TensorFlow 2.2 หรือใหม่กว่า
  • pip 19.0 ขึ้นไป (ต้องการการรองรับ manylinux2010 )
  • Ubuntu 16.04 หรือใหม่กว่า (64 บิต)
  • macOS 10.12.6 (Sierra) หรือใหม่กว่า (64-bit) (ไม่รองรับ GPU)
    • macOS ต้องใช้ pip 20.3 ขึ้นไป
  • Windows 7 หรือใหม่กว่า (64 บิต)
  • รองรับ GPU ต้องใช้การ์ดที่เปิดใช้งานCUDA® (Ubuntu และ Windows)

ข้อกำหนดฮาร์ดแวร์

1. ติดตั้งสภาพแวดล้อมการพัฒนา Python บนระบบของคุณ

ตรวจสอบว่าสภาพแวดล้อม Python ของคุณได้รับการกำหนดค่าไว้แล้วหรือไม่:

python3 --version
pip3 --version

หากติดตั้งแพ็กเกจเหล่านี้แล้วให้ข้ามไปขั้นตอนถัดไป
มิฉะนั้นให้ติดตั้ง Python ตัว จัดการแพ็คเกจ pip และ venv :

Ubuntu

sudo apt update
sudo apt install python3-dev python3-pip python3-venv

macOS

ติดตั้งโดยใช้ตัวจัดการแพ็คเกจ Homebrew :

/usr/bin/ruby -e "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/master/install)"
export PATH="/usr/local/opt/python/libexec/bin:$PATH"
# if you are on macOS 10.12 (Sierra) use `export PATH="/usr/local/bin:/usr/local/sbin:$PATH"`
brew update
brew install python  # Python 3

Windows

ติดตั้ง Microsoft Visual C ++ Redistributable สำหรับ Visual Studio 2015, 2017 และ 2019 ตั้งแต่รุ่น TensorFlow 2.1.0 ไฟล์ msvcp140_1.dll จำเป็นต้องใช้จากแพ็คเกจนี้ (ซึ่งอาจไม่ได้ให้มาจากแพ็คเกจที่แจกจ่ายต่อได้รุ่นเก่า) แจกจ่ายซ้ำได้มาพร้อมกับ Visual Studio 2019 แต่สามารถติดตั้งแยกต่างหากได้:

  1. ไปที่ดาวน์โหลด Microsoft Visual C ++
  2. เลื่อนหน้าลงไปที่ส่วน Visual Studio 2015, 2017 และ 2019
  3. ดาวน์โหลดและติดตั้ง Microsoft Visual C ++ Redistributable สำหรับ Visual Studio 2015, 2017 และ 2019 สำหรับแพลตฟอร์มของคุณ

ตรวจสอบให้แน่ใจว่า เส้นทางยาวเปิดใช้งาน บน Windows

ติดตั้ง Python 3 รุ่น 64 บิต สำหรับ Windows (เลือก pip เป็นคุณสมบัติเสริม)

อื่น ๆ

curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py
python get-pip.py

สภาพแวดล้อมเสมือน Python ใช้เพื่อแยกการติดตั้งแพ็คเกจออกจากระบบ

Ubuntu / macOS

สร้างสภาพแวดล้อมเสมือนใหม่โดยเลือกล่าม Python และสร้างไดเร็กทอรี. / ./venv เพื่อเก็บไว้:

python3 -m venv --system-site-packages ./venv

เปิดใช้งานสภาพแวดล้อมเสมือนโดยใช้คำสั่งเฉพาะเชลล์:

source ./venv/bin/activate  # sh, bash, or zsh
. ./venv/bin/activate.fish  # fish
source ./venv/bin/activate.csh  # csh or tcsh

เมื่อสภาพแวดล้อมเสมือนแอ็คทีฟเชลล์พร้อมต์ของคุณจะถูกนำหน้าด้วย (venv)

ติดตั้งแพ็กเกจภายในสภาพแวดล้อมเสมือนโดยไม่มีผลต่อการตั้งค่าระบบโฮสต์ เริ่มต้นด้วยการอัพเกรด pip :

pip install --upgrade pip

pip list  # show packages installed within the virtual environment

และเพื่อออกจากสภาพแวดล้อมเสมือนในภายหลัง:

deactivate  # don't exit until you're done using TensorFlow

Windows

สร้างสภาพแวดล้อมเสมือนใหม่โดยเลือกล่าม Python และสร้างไดเร็กทอรี .\venv เพื่อเก็บไว้:

python -m venv --system-site-packages .\venv

เปิดใช้งานสภาพแวดล้อมเสมือน:

.\venv\Scripts\activate

ติดตั้งแพ็กเกจภายในสภาพแวดล้อมเสมือนโดยไม่มีผลต่อการตั้งค่าระบบโฮสต์ เริ่มต้นด้วยการอัพเกรด pip :

pip install --upgrade pip

pip list  # show packages installed within the virtual environment

และเพื่อออกจากสภาพแวดล้อมเสมือนในภายหลัง:

deactivate  # don't exit until you're done using TensorFlow

Conda

แม้ว่า TensorFlow จะแนะนำแพ็คเกจ pip แต่ ก็มี แพ็คเกจ Anaconda ที่ ชุมชนรองรับ หากต้องการติดตั้งโปรดอ่านคู่มือ Anaconda TensorFlow

3. ติดตั้งแพ็คเกจ pip TensorFlow

เลือกหนึ่งในแพ็คเกจ TensorFlow ต่อไปนี้เพื่อติดตั้ง จาก PyPI :

  • tensorflow - รุ่นล่าสุดที่เสถียรพร้อมรองรับ CPU และ GPU (Ubuntu และ Windows)
  • tf-nightly สร้างตัวอย่าง (ไม่เสถียร) Ubuntu และ Windows รวมถึง การรองรับ GPU
  • tensorflow==1.15 - เวอร์ชันสุดท้ายของ TensorFlow 1.x.

ติดตั้งสภาพแวดล้อมเสมือน

pip install --upgrade tensorflow

ตรวจสอบการติดตั้ง:

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

ติดตั้งระบบ

pip3 install --user --upgrade tensorflow  # install in $HOME

ตรวจสอบการติดตั้ง:

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

สถานที่บรรจุภัณฑ์

กลไกการติดตั้งบางอย่างต้องการ URL ของแพ็คเกจ TensorFlow Python ค่าที่คุณระบุขึ้นอยู่กับเวอร์ชัน Python ของคุณ

เวอร์ชัน URL
ลินุกซ์
รองรับ Python 3.6 GPU https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-2.5.0-cp36-cp36m-manylinux2010_x86_64.whl
Python 3.6 CPU เท่านั้น https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow_cpu-2.5.0-cp36-cp36m-manylinux2010_x86_64.whl
รองรับ Python 3.7 GPU https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-2.5.0-cp37-cp37m-manylinux2010_x86_64.whl
Python 3.7 CPU เท่านั้น https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow_cpu-2.5.0-cp37-cp37m-manylinux2010_x86_64.whl
รองรับ Python 3.8 GPU https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-2.5.0-cp38-cp38-manylinux2010_x86_64.whl
Python 3.8 CPU เท่านั้น https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow_cpu-2.5.0-cp38-cp38-manylinux2010_x86_64.whl
รองรับ Python 3.9 GPU https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-2.5.0-cp39-cp39-manylinux2010_x86_64.whl
Python 3.9 CPU เท่านั้น https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow_cpu-2.5.0-cp39-cp39-manylinux2010_x86_64.whl
macOS (เฉพาะ CPU)
Python 3.6 https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-2.5.0-cp36-cp36m-macosx_10_11_x86_64.whl
Python 3.7 https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-2.5.0-cp37-cp37m-macosx_10_11_x86_64.whl
Python 3.8 https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-2.5.0-cp38-cp38-macosx_10_11_x86_64.whl
Python 3.8 https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-2.5.0-cp39-cp39-macosx_10_11_x86_64.whl
Windows
รองรับ Python 3.6 GPU https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-2.5.0-cp36-cp36m-win_amd64.whl
Python 3.6 CPU เท่านั้น https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow_cpu-2.5.0-cp36-cp36m-win_amd64.whl
รองรับ Python 3.7 GPU https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-2.5.0-cp37-cp37m-win_amd64.whl
Python 3.7 CPU เท่านั้น https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow_cpu-2.5.0-cp37-cp37m-win_amd64.whl
รองรับ Python 3.8 GPU https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-2.5.0-cp38-cp38-win_amd64.whl
Python 3.8 CPU เท่านั้น https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow_cpu-2.5.0-cp38-cp38-win_amd64.whl
รองรับ Python 3.9 GPU https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-2.5.0-cp39-cp39-win_amd64.whl
Python 3.9 CPU เท่านั้น https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow_cpu-2.5.0-cp39-cp39-win_amd64.whl