RSVP for your your local TensorFlow Everywhere event today!
Trang này được dịch bởi Cloud Translation API.
Switch to English

Cài đặt TensorFlow bằng pip

TensorFlow 2 gói có sẵn

  • tensorflow phát hành ổn định nhất với hỗ trợ CPU và GPU (Ubuntu và Windows)
  • tf-nightly —Xem trước bản dựng (không ổn định) . Ubuntu và Windows bao gồm hỗ trợ GPU .

Phiên bản cũ hơn của TensorFlow

Đối với TensorFlow 1.x, các gói CPU và GPU là riêng biệt:

  • tensorflow==1.15 —Release chỉ dành cho CPU
  • tensorflow-gpu==1.15 —Tăng tốc với hỗ trợ GPU (Ubuntu và Windows)

Yêu cầu hệ thống

Yêu cầu phần cứng

1. Cài đặt môi trường phát triển Python trên hệ thống của bạn

Kiểm tra xem môi trường Python của bạn đã được định cấu hình chưa:

python3 --version
pip3 --version

Nếu các gói này đã được cài đặt, hãy chuyển sang bước tiếp theo.
Nếu không, hãy cài đặt Python , trình quản lý gói pipvenv :

Ubuntu

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

hệ điều hành Mac

Cài đặt bằng trình quản lý gói 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

các cửa sổ

Cài đặt Microsoft Visual C ++ Redistributable cho Visual Studio 2015, 2017 và 2019 . Bắt đầu với phiên bản TensorFlow 2.1.0, tệp msvcp140_1.dll là bắt buộc từ gói này (có thể không được cung cấp từ các gói có thể phân phối lại cũ hơn). Có thể phân phối lại đi kèm với Visual Studio 2019 nhưng có thể được cài đặt riêng:

  1. Đi tới phần tải xuống Microsoft Visual C ++ ,
  2. Cuộn xuống trang đến phần Visual Studio 2015, 2017 và 2019 .
  3. Tải xuống và cài đặt Microsoft Visual C ++ Redistributable cho Visual Studio 2015, 2017 và 2019 cho nền tảng của bạn.

Đảm bảo rằng các đường dẫn dài được bật trên Windows.

Cài đặt bản phát hành Python 3 64-bit cho Windows (chọn pip làm tính năng tùy chọn).

Raspberry Pi

Yêu cầu đối với hệ điều hành Raspbian :

sudo apt update
sudo apt install python3-dev python3-pip python3-venv
sudo apt install libatlas-base-dev        # required for numpy

Khác

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

Môi trường ảo Python được sử dụng để cách ly cài đặt gói khỏi hệ thống.

Ubuntu / macOS

Tạo một môi trường ảo mới bằng cách chọn một trình thông dịch Python và tạo một thư mục ./venv để chứa nó:

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

Kích hoạt môi trường ảo bằng lệnh shell cụ thể:

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

Khi môi trường ảo đang hoạt động, dấu nhắc trình bao của bạn có tiền tố là (venv) .

Cài đặt các gói trong môi trường ảo mà không ảnh hưởng đến thiết lập hệ thống máy chủ. Bắt đầu bằng cách nâng cấp pip :

pip install --upgrade pip

pip list  # show packages installed within the virtual environment

Và để thoát khỏi môi trường ảo sau này:

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

các cửa sổ

Tạo một môi trường ảo mới bằng cách chọn một trình thông dịch Python và tạo một thư mục .\venv để chứa nó:

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

Kích hoạt môi trường ảo:

.\venv\Scripts\activate

Cài đặt các gói trong môi trường ảo mà không ảnh hưởng đến thiết lập hệ thống máy chủ. Bắt đầu bằng cách nâng cấp pip :

pip install --upgrade pip

pip list  # show packages installed within the virtual environment

Và để thoát khỏi môi trường ảo sau này:

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

Conda

Mặc dù gói pip do TensorFlow cung cấp được khuyến nghị, nhưng gói Anaconda được cộng đồng hỗ trợ vẫn có sẵn. Để cài đặt, hãy đọc hướng dẫn Anaconda TensorFlow .

3. Cài đặt gói pip TensorFlow

Chọn một trong các gói TensorFlow sau để cài đặt từ PyPI :

  • tensorflow phát hành ổn định nhất với hỗ trợ CPU và GPU (Ubuntu và Windows) .
  • tf-nightly —Xem trước bản dựng (không ổn định) . Ubuntu và Windows bao gồm hỗ trợ GPU .
  • tensorflow==1.15 — Phiên bản cuối cùng của TensorFlow 1.x.

Cài đặt môi trường ảo

pip install --upgrade tensorflow

Xác minh cài đặt:

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

Cài đặt hệ thống

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

Xác minh cài đặt:

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

Vị trí gói hàng

Một số cơ chế cài đặt yêu cầu URL của gói TensorFlow Python. Giá trị bạn chỉ định phụ thuộc vào phiên bản Python của bạn.

Phiên bản URL
Linux
Hỗ trợ GPU Python 3.6 https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-2.4.0-cp36-cp36m-manylinux2010_x86_64.whl
Python 3.6 chỉ dành cho CPU https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow_cpu-2.4.0-cp36-cp36m-manylinux2010_x86_64.whl
Hỗ trợ GPU Python 3.7 https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-2.4.0-cp37-cp37m-manylinux2010_x86_64.whl
Python 3.7 chỉ dành cho CPU https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow_cpu-2.4.0-cp37-cp37m-manylinux2010_x86_64.whl
Hỗ trợ GPU Python 3.8 https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-2.4.0-cp38-cp38-manylinux2010_x86_64.whl
Python 3.8 chỉ dành cho CPU https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow_cpu-2.4.0-cp38-cp38-manylinux2010_x86_64.whl
macOS (chỉ dành cho CPU)
Python 3.6 https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-2.4.0-cp36-cp36m-macosx_10_9_x86_64.whl
Python 3.7 https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-2.4.0-cp37-cp37m-macosx_10_9_x86_64.whl
Python 3,8 https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-2.4.0-cp38-cp38-macosx_10_14_x86_64.whl
các cửa sổ
Hỗ trợ GPU Python 3.6 https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-2.4.0-cp36-cp36m-win_amd64.whl
Python 3.6 chỉ dành cho CPU https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow_cpu-2.4.0-cp36-cp36m-win_amd64.whl
Hỗ trợ GPU Python 3.7 https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-2.4.0-cp37-cp37m-win_amd64.whl
Python 3.7 chỉ dành cho CPU https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow_cpu-2.4.0-cp37-cp37m-win_amd64.whl
Hỗ trợ GPU Python 3.8 https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-2.4.0-cp38-cp38-win_amd64.whl
Python 3.8 chỉ dành cho CPU https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow_cpu-2.4.0-cp38-cp38-win_amd64.whl
Raspberry PI (chỉ dành cho CPU)
Python 3, Pi0 hoặc Pi1 https://storage.googleapis.com/tensorflow/raspberrypi/tensorflow-2.3.0rc2-cp35-none-linux_armv6l.whl
Python 3, Pi2 hoặc Pi3 https://storage.googleapis.com/tensorflow/raspberrypi/tensorflow-2.3.0rc2-cp35-none-linux_armv6l.whl