Google cam kết thúc đẩy công bằng chủng tộc cho các cộng đồng Đen. Xem cách làm.
Trang này được dịch bởi Cloud Translation API.
Switch to English

Xây dựng từ nguồn trên Windows

Xây dựng gói pip TensorFlow từ nguồn và cài đặt nó trên Windows.

Thiết lập cho Windows

Cài đặt các công cụ xây dựng sau để cấu hình môi trường phát triển Windows của bạn.

Cài đặt phụ thuộc gói Python và TensorFlow

Cài đặt bản phát hành Python 3.5.x hoặc Python 3.6.x 64 bit cho Windows . Chọn pip làm tính năng tùy chọn và thêm nó vào biến môi trường %PATH% của bạn.

Cài đặt các phụ thuộc gói pip TensorFlow:

pip3 install six numpy wheel
pip3 install keras_applications==1.0.6 --no-deps
pip3 install keras_preprocessing==1.0.5 --no-deps

Các phần phụ thuộc được liệt kê trong tệp setup.py trong REQUIRED_PACKAGES .

Cài đặt Bazel

Cài đặt Bazel , công cụ xây dựng được sử dụng để biên dịch TensorFlow. Đối với phiên bản Bazel, hãy xem cấu hình bản dựng được thử nghiệm cho Windows. Cấu hình Bazel để xây dựng C ++ .

Thêm vị trí của tệp thực thi Bazel vào biến môi trường %PATH% của bạn.

Cài đặt MSYS2

Cài đặt MSYS2 cho các công cụ bin cần thiết để xây dựng TensorFlow. Nếu MSYS2 được cài đặt vào C:\msys64 , hãy thêm C:\msys64\usr\bin vào biến môi trường %PATH% của bạn. Sau đó, sử dụng cmd.exe , chạy:

pacman -S git patch unzip

Cài đặt công cụ xây dựng Visual C ++ 2019

Cài đặt công cụ xây dựng Visual C ++ 2019 . Điều này đi kèm với Visual Studio 2019 nhưng có thể được cài đặt riêng:

  1. Chuyển đến phần tải xuống của Visual Studio ,
  2. Chọn Redistributables và Build Tools ,
  3. Tải xuống và cài đặt:
    • Microsoft Visual C ++ 2019 có thể phân phối lại
    • Công cụ xây dựng Microsoft 2019

Cài đặt hỗ trợ GPU (tùy chọn)

Xem hướng dẫn hỗ trợ GPU của Windows để cài đặt trình điều khiển và phần mềm bổ sung cần thiết để chạy TensorFlow trên GPU.

Tải xuống mã nguồn TensorFlow

Sử dụng Git để sao chép kho lưu trữ TensorFlow ( git được cài đặt với MSYS2):

git clone https://github.com/tensorflow/tensorflow.git
cd tensorflow

Các repo mặc định cho nhánh phát triển master . Bạn cũng có thể kiểm tra một nhánh phát hành để xây dựng:

git checkout branch_name  # r1.9, r1.10, etc.

Cấu hình bản dựng

Định cấu hình xây dựng hệ thống của bạn bằng cách chạy phần sau ở gốc của cây nguồn TensorFlow của bạn:

python ./configure.py

Kịch bản lệnh này nhắc bạn về vị trí của các phụ thuộc TensorFlow và yêu cầu các tùy chọn cấu hình xây dựng bổ sung (ví dụ: cờ trình biên dịch). Phần sau đây cho thấy một lần chạy mẫu của python ./configure.py (phiên của bạn có thể khác nhau):

Tùy chọn cấu hình

Để hỗ trợ GPU , chỉ định các phiên bản CUDA và cuDNN. Nếu hệ thống của bạn có nhiều phiên bản CUDA hoặc cuDNN được cài đặt, hãy đặt rõ ràng phiên bản thay vì dựa vào mặc định. ./configure.py tạo các liên kết tượng trưng đến các thư viện CUDA của hệ thống của bạn, vì vậy nếu bạn cập nhật đường dẫn thư viện CUDA của mình, bước cấu hình này phải được chạy lại trước khi xây dựng.

Xây dựng gói pip

Kéo căng 2.x

tenorflow: master repo đã được cập nhật để xây dựng 2.x theo mặc định. Cài đặt Bazel và sử dụng bazel build để tạo gói TensorFlow.

bazel build //tensorflow/tools/pip_package:build_pip_package

Kéo căng 1.x

Để xây dựng phiên bản 1.x của TensorFlow từ bản gốc, hãy sử dụng bazel build --config=v1 để tạo gói TensorFlow 1.x.

bazel build --config=v1 //tensorflow/tools/pip_package:build_pip_package

Chỉ dành cho CPU

Sử dụng bazel để tạo trình xây dựng gói TensorFlow với sự hỗ trợ chỉ dành cho CPU:

bazel build --config=opt //tensorflow/tools/pip_package:build_pip_package

Hỗ trợ GPU

Để tạo trình xây dựng gói TensorFlow có hỗ trợ GPU:

bazel build --config=opt --config=cuda --define=no_tensorflow_py_deps=true //tensorflow/tools/pip_package:build_pip_package

Tùy chọn xây dựng Bazel

Sử dụng tùy chọn này khi xây dựng để tránh sự cố với việc tạo gói: tenorflow: vấn đề # 22390

--define=no_tensorflow_py_deps=true

Xem tham chiếu dòng lệnh của Bazel để biết các tùy chọn xây dựng .

Xây dựng TensorFlow từ nguồn có thể sử dụng rất nhiều RAM. Nếu hệ thống của bạn bị giới hạn bộ nhớ, hãy giới hạn mức sử dụng RAM của Bazel với: --local_ram_resources=2048 .

Nếu xây dựng với sự hỗ trợ GPU, hãy thêm --copt=-nvcc_options=disable-warnings để chặn các thông báo cảnh báo nvcc.

Xây dựng gói

bazel build tạo ra một tệp thực thi có tên build_pip_package là chương trình xây dựng gói pip . Ví dụ: phần sau đây xây dựng gói .whl trong C:/tmp/tensorflow_pkg :

bazel-bin\tensorflow\tools\pip_package\build_pip_package C:/tmp/tensorflow_pkg

Mặc dù có thể xây dựng cả cấu hình CUDA và không phải CUDA trong cùng một cây nguồn, chúng tôi khuyên bạn nên chạy bazel clean khi chuyển đổi giữa hai cấu hình này trong cùng một cây nguồn.

Cài đặt gói

Tên tệp của tệp .whl được tạo tùy thuộc vào phiên bản TensorFlow và nền tảng của bạn. Sử dụng pip3 install để cài đặt gói, ví dụ:

pip3 install C:/tmp/tensorflow_pkg/tensorflow- version -cp36-cp36m-win_amd64.whl

Xây dựng bằng trình bao MSYS

TensorFlow cũng có thể được xây dựng bằng cách sử dụng trình bao MSYS. Thực hiện các thay đổi được liệt kê bên dưới, sau đó làm theo các hướng dẫn trước đó cho dòng lệnh gốc của Windows ( cmd.exe ).

Vô hiệu hóa chuyển đổi đường dẫn MSYS

MSYS tự động chuyển đổi các đối số trông giống như các đường dẫn Unix sang các đường dẫn Windows và điều này không hoạt động với bazel . (Nhãn //path/to:bin được coi là đường dẫn tuyệt đối Unix vì nó bắt đầu bằng dấu gạch chéo.)

export MSYS_NO_PATHCONV=1
export MSYS2_ARG_CONV_EXCL="*"

Đặt ĐƯỜNG của bạn

Thêm các thư mục cài đặt Bazel và Python vào biến môi trường $PATH của bạn. Nếu Bazel được cài đặt thành C:\tools\bazel.exe và Python thành C:\Python36\python.exe , hãy đặt PATH của bạn với:

# Use Unix-style with ':' as separator
export PATH="/c/tools:$PATH"
export PATH="/c/Python36:$PATH"

Để được hỗ trợ GPU, hãy thêm các thư mục bin CUDA và cuDNN vào $PATH của bạn:

export PATH="/c/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v9.0/bin:$PATH"
export PATH="/c/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v9.0/extras/CUPTI/libx64:$PATH"
export PATH="/c/tools/cuda/bin:$PATH"

Đã kiểm tra cấu hình bản dựng

CPU

Phiên bản Phiên bản Python Trình biên dịch Xây dựng công cụ
dòng chảy-2.3.0 3.5-3.8 MSVC 2019 Bazel 3.1.0
dòng chảy-2.2.0 3.5-3.8 MSVC 2019 Bazel 2.0.0
tenorflow-2.1.0 3.5-3.7 MSVC 2019 Bazel 0,27,1-0,29.1
tenorflow-2.0.0 3.5-3.7 MSVC 2017 Bazel 0,26.1
dòng chảy-1.15.0 3.5-3.7 MSVC 2017 Bazel 0,26.1
tenorflow-1.14.0 3.5-3.7 MSVC 2017 Bazel 0,24.1-0.25.2
dòng chảy-1.13.0 3.5-3.7 Cập nhật MSVC 2015 3 Bazel 0.19.0-0.21.0
dòng chảy-1.12.0 3.5-3.6 Cập nhật MSVC 2015 3 Bazel 0.15.0
tenorflow-1.11.0 3.5-3.6 Cập nhật MSVC 2015 3 Bazel 0.15.0
dòng chảy-1.10.0 3.5-3.6 Cập nhật MSVC 2015 3 Cmake v3.6.3
dòng chảy-1.9.0 3.5-3.6 Cập nhật MSVC 2015 3 Cmake v3.6.3
dòng chảy-1.8.0 3.5-3.6 Cập nhật MSVC 2015 3 Cmake v3.6.3
dòng chảy-1.7.0 3.5-3.6 Cập nhật MSVC 2015 3 Cmake v3.6.3
dòng chảy-1.6.0 3.5-3.6 Cập nhật MSVC 2015 3 Cmake v3.6.3
dòng chảy-1.5.0 3.5-3.6 Cập nhật MSVC 2015 3 Cmake v3.6.3
dòng chảy-1.4.0 3.5-3.6 Cập nhật MSVC 2015 3 Cmake v3.6.3
dòng chảy-1.3.0 3.5-3.6 Cập nhật MSVC 2015 3 Cmake v3.6.3
tenorflow-1.2.0 3.5-3.6 Cập nhật MSVC 2015 3 Cmake v3.6.3
dòng chảy-1.1.0 3,5 Cập nhật MSVC 2015 3 Cmake v3.6.3
tenorflow-1.0.0 3,5 Cập nhật MSVC 2015 3 Cmake v3.6.3

GPU

Phiên bản Phiên bản Python Trình biên dịch Xây dựng công cụ cuDNN CUDA
tenorflow_gpu-2.3.0 3.5-3.8 MSVC 2019 Bazel 3.1.0 7.4 10.1
tenorflow_gpu-2.2.0 3.5-3.8 MSVC 2019 Bazel 2.0.0 7.4 10.1
tenorflow_gpu-2.1.0 3.5-3.7 MSVC 2019 Bazel 0,27,1-0,29.1 7.4 10.1
tenorflow_gpu-2.0.0 3.5-3.7 MSVC 2017 Bazel 0,26.1 7.4 10
tenorflow_gpu-1.15.0 3.5-3.7 MSVC 2017 Bazel 0,26.1 7.4 10
tenorflow_gpu-1.14.0 3.5-3.7 MSVC 2017 Bazel 0,24.1-0.25.2 7.4 10
tenorflow_gpu-1.13.0 3.5-3.7 Cập nhật MSVC 2015 3 Bazel 0.19.0-0.21.0 7.4 10
tenorflow_gpu-1.12.0 3.5-3.6 Cập nhật MSVC 2015 3 Bazel 0.15.0 7 9
tenorflow_gpu-1.11.0 3.5-3.6 Cập nhật MSVC 2015 3 Bazel 0.15.0 7 9
tenorflow_gpu-1.10.0 3.5-3.6 Cập nhật MSVC 2015 3 Cmake v3.6.3 7 9
tenorflow_gpu-1.9.0 3.5-3.6 Cập nhật MSVC 2015 3 Cmake v3.6.3 7 9
tenorflow_gpu-1.8.0 3.5-3.6 Cập nhật MSVC 2015 3 Cmake v3.6.3 7 9
tenorflow_gpu-1.7.0 3.5-3.6 Cập nhật MSVC 2015 3 Cmake v3.6.3 7 9
tenorflow_gpu-1.6.0 3.5-3.6 Cập nhật MSVC 2015 3 Cmake v3.6.3 7 9
tenorflow_gpu-1.5.0 3.5-3.6 Cập nhật MSVC 2015 3 Cmake v3.6.3 7 9
tenorflow_gpu-1.4.0 3.5-3.6 Cập nhật MSVC 2015 3 Cmake v3.6.3 6 số 8
tenorflow_gpu-1.3.0 3.5-3.6 Cập nhật MSVC 2015 3 Cmake v3.6.3 6 số 8
tenorflow_gpu-1.2.0 3.5-3.6 Cập nhật MSVC 2015 3 Cmake v3.6.3 5.1 số 8
tenorflow_gpu-1.1.0 3,5 Cập nhật MSVC 2015 3 Cmake v3.6.3 5.1 số 8
tenorflow_gpu-1.0.0 3,5 Cập nhật MSVC 2015 3 Cmake v3.6.3 5.1 số 8