สร้างจากแหล่งที่มาบน Windows

สร้างแพ็คเกจ TensorFlow pip จากซอร์สและติดตั้งบน Windows

การตั้งค่าสำหรับ Windows

ติดตั้งเครื่องมือสร้างต่อไปนี้เพื่อกำหนดค่าสภาพแวดล้อมการพัฒนา Windows ของคุณ

ติดตั้ง Python และการอ้างอิงแพ็คเกจ TensorFlow

ติดตั้ง Python 3.9+ รุ่น 64 บิตสำหรับ Windows เลือก pip เป็นคุณสมบัติเสริมและเพิ่มไปยังตัวแปรสภาพแวดล้อม %PATH% ของคุณ

ติดตั้งการพึ่งพาแพ็คเกจ TensorFlow pip :

pip3 install -U six numpy wheel packaging
pip3 install -U keras_preprocessing --no-deps

การขึ้นต่อกันจะแสดงอยู่ในไฟล์ setup.py ภายใต้ REQUIRED_PACKAGES

ติดตั้ง Bazel

ติดตั้ง Bazel ซึ่งเป็นเครื่องมือสร้างที่ใช้ในการรวบรวม TensorFlow สำหรับเวอร์ชัน Bazel โปรดดู การกำหนดค่าบิลด์ที่ทดสอบแล้ว สำหรับ Windows กำหนดค่า Bazel เพื่อ สร้าง C++

เพิ่มตำแหน่งของโปรแกรมเรียกทำงาน Bazel ให้กับตัวแปรสภาพแวดล้อม %PATH% ของคุณ

ติดตั้ง MSYS2

ติดตั้ง MSYS2 สำหรับเครื่องมือ bin ที่จำเป็นในการสร้าง TensorFlow หากติดตั้ง MSYS2 ไว้ที่ C:\msys64 ให้เพิ่ม C:\msys64\usr\bin ไปยังตัวแปรสภาพแวดล้อม %PATH% ของคุณ จากนั้นใช้ cmd.exe เรียกใช้:

pacman -S git patch unzip

ติดตั้งเครื่องมือสร้าง Visual C++ 2019

ติดตั้ง เครื่องมือสร้าง Visual C++ 2019 สิ่งนี้มาพร้อมกับ Visual Studio 2019 แต่สามารถติดตั้งแยกต่างหาก:

1. ไปที่ การดาวน์โหลด Visual Studio
2. เลือก Redistributables และสร้างเครื่องมือ
3. ดาวน์โหลดและติดตั้ง:
   • Microsoft Visual C++ 2019 เผยแพร่ต่อได้
   • เครื่องมือสร้าง Microsoft 2019

ติดตั้งการสนับสนุน GPU (ไม่บังคับ)

ดูคู่มือ การสนับสนุน Windows GPU เพื่อติดตั้งไดรเวอร์และซอฟต์แวร์เพิ่มเติมที่จำเป็นสำหรับการเรียกใช้ TensorFlow บน GPU

ดาวน์โหลดซอร์สโค้ด TensorFlow

ใช้ Git เพื่อโคลน ที่เก็บ TensorFlow (ติดตั้ง git กับ MSYS2):

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

repo เริ่มต้นที่สาขาการพัฒนา master คุณยังสามารถตรวจสอบ สาขาการเผยแพร่ เพื่อสร้าง:

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

ทางเลือก: การตั้งค่าตัวแปรสภาพแวดล้อม

เรียกใช้คำสั่งต่อไปนี้ก่อนที่จะรันคำสั่ง build เพื่อหลีกเลี่ยงปัญหาเกี่ยวกับการสร้างแพ็คเกจ: (หากตั้งค่าคำสั่งด้านล่างขณะติดตั้งแพ็คเกจ โปรดอย่าสนใจ) เรียกใช้ set ตรวจสอบว่าเส้นทางทั้งหมดถูกตั้งค่าอย่างถูกต้องหรือไม่ เรียกใช้ echo %Environmental Variable% เช่น echo %BAZEL_VC% เพื่อตรวจสอบเส้นทางที่ตั้งค่าสำหรับตัวแปรสภาพแวดล้อมเฉพาะ

เส้นทาง Python ตั้งค่าปัญหา tensorflow:issue#59943 , tensorflow:issue#9436 , tensorflow:issue#60083

set PATH=path/to/python # [e.g. (C:/Python310)]
set PATH=path/to/python/Scripts # [e.g. (C:/Python310/Scripts)] 
set PYTHON_BIN_PATH=path/to/python_virtualenv/Scripts/python.exe 
set PYTHON_LIB_PATH=path/to/python virtualenv/lib/site-packages 
set PYTHON_DIRECTORY=path/to/python_virtualenv/Scripts 

เส้นทาง Bazel/MSVC ตั้งค่าปัญหา tensorflow:issue#54578

set BAZEL_SH=C:/msys64/usr/bin/bash.exe 
set BAZEL_VS=C:/Program Files(x86)/Microsoft Visual Studio/2019/BuildTools 
set BAZEL_VC=C:/Program Files(x86)/Microsoft Visual Studio/2019/BuildTools/VC 

ทางเลือก: กำหนดค่าการสร้าง

การสร้าง TensorFlow ได้รับการกำหนดค่าโดยไฟล์ .bazelrc ในไดเร็กทอรีรากของที่เก็บ สามารถใช้สคริปต์ ./configure หรือ ./configure.py เพื่อปรับการตั้งค่าทั่วไป

หากคุณต้องการเปลี่ยนคอนฟิกูเรชัน ให้รันสคริปต์ ./configure จากไดเร็กทอรีรูทของที่เก็บ

python ./configure.py

สคริปต์นี้แจ้งให้คุณทราบตำแหน่งของการพึ่งพา TensorFlow และขอตัวเลือกการกำหนดค่าบิลด์เพิ่มเติม (เช่น แฟล็กคอมไพลเลอร์) ต่อไปนี้แสดงตัวอย่างการทำงานของ python ./configure.py (เซสชันของคุณอาจแตกต่างออกไป):

python ./configure.py
You have bazel 5.3.0 installed.
Please specify the location of python. [Default is C:\Python310\python.exe]:
Found possible Python library paths:
C:\Python310\lib\site-packages
Please input the desired Python library path to use.  Default is [C:\Python310\lib\site-packages]

Do you wish to build TensorFlow with ROCm support? [y/N]:
No ROCm support will be enabled for TensorFlow.


Warning: Cannot build with CUDA support on Windows.
Starting in TF 2.11, CUDA build is not supported for Windows. For using TensorFlow GPU on Windows, you will need to build/install TensorFlow in WSL2.

Please specify optimization flags to use during compilation when bazel option "--config=opt" is specified [Default is /arch:AVX]:


Would you like to override eigen strong inline for some C++ compilation to reduce the compilation time? [Y/n]:
Eigen strong inline overridden.

Would you like to interactively configure ./WORKSPACE for Android builds? [y/N]:
Not configuring the WORKSPACE for Android builds.

Preconfigured Bazel build configs. You can use any of the below by adding "--config=<>" to your build command. See .bazelrc for more details.
        --config=mkl            # Build with MKL support.
        --config=mkl_aarch64    # Build with oneDNN and Compute Library for the Arm Architecture (ACL).
        --config=monolithic     # Config for mostly static monolithic build.
        --config=numa           # Build with NUMA support.
        --config=dynamic_kernels        # (Experimental) Build kernels into separate shared objects.
        --config=v1             # Build with TensorFlow 1 API instead of TF 2 API.
Preconfigured Bazel build configs to DISABLE default on features:
        --config=nogcp          # Disable GCP support.
        --config=nonccl         # Disable NVIDIA NCCL support.

สร้างและติดตั้งแพ็คเกจ pip

แพ็คเกจ pip สร้างขึ้นในสองขั้นตอน คำสั่ง bazel build สร้างโปรแกรม "ตัวสร้างแพ็คเกจ" จากนั้นคุณเรียกใช้ตัวสร้างแพ็คเกจเพื่อสร้างแพ็คเกจ

สร้างตัวสร้างแพ็คเกจ

tensorflow:master repo ได้รับการอัปเดตเป็น build 2.x ตามค่าเริ่มต้น ติดตั้ง Bazel และใช้ bazel build เพื่อสร้าง TensorFlow package-builder

bazel build //tensorflow/tools/pip_package:build_pip_package

CPU เท่านั้น

ใช้ bazel เพื่อสร้างตัวสร้างแพ็คเกจ TensorFlow ที่รองรับ CPU เท่านั้น:

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

รองรับ GPU

ในการสร้างตัวสร้างแพ็คเกจ TensorFlow พร้อมการรองรับ GPU:

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

คำสั่งล้างแคช bazel เพื่อแก้ไขข้อผิดพลาดเนื่องจากข้อมูลที่แคชไม่ถูกต้องหรือล้าสมัย bazel clean ด้วย --expunge flag ลบไฟล์อย่างถาวร

bazel clean 
bazel clean --expunge  

ตัวเลือกการสร้าง Bazel

ใช้ตัวเลือกนี้เมื่อสร้างเพื่อหลีกเลี่ยงปัญหาเกี่ยวกับการสร้างแพ็คเกจ: tensorflow:issue#22390

--define=no_tensorflow_py_deps=true

ดู การอ้างอิงบรรทัดคำสั่ง Bazel สำหรับ ตัวเลือกการสร้าง

การสร้าง TensorFlow จากแหล่งที่มาสามารถใช้ RAM จำนวนมากได้ หากระบบของคุณจำกัดหน่วยความจำ ให้จำกัดการใช้ RAM ของ Bazel ด้วย: --local_ram_resources=2048

หากสร้างด้วยการรองรับ GPU ให้เพิ่ม --copt=-nvcc_options=disable-warnings เพื่อระงับข้อความเตือน nvcc

สร้างแพ็คเกจ

คำสั่ง bazel build สร้างไฟล์เรียกทำงานชื่อ build_pip_package ซึ่งเป็นโปรแกรมที่สร้างแพ็คเกจ pip ตัวอย่างเช่น ต่อไปนี้สร้างแพ็คเกจ .whl ในไดเร็กทอรี C:/tmp/tensorflow_pkg :

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

แม้ว่าจะสามารถสร้างทั้งการกำหนดค่า CUDA และไม่ใช่ CUDA ภายใต้แผนผังต้นทางเดียวกัน เราขอแนะนำให้เรียกใช้ bazel clean เมื่อสลับระหว่างการกำหนดค่าทั้งสองนี้ในแผนผังต้นทางเดียวกัน

ติดตั้งแพ็คเกจ

ชื่อไฟล์ของไฟล์ .whl ที่สร้างขึ้นนั้นขึ้นอยู่กับเวอร์ชัน TensorFlow และแพลตฟอร์มของคุณ ใช้ pip3 install เพื่อติดตั้งแพ็คเกจ เช่น:

pip3 install C:/tmp/tensorflow_pkg/tensorflow-version-tags.whl

e.g. pip3 install C:/tmp/tensorflow_pkg/tensorflow-2.12.0-cp310-cp310-win_amd64.whl

สร้างโดยใช้เชลล์ MSYS

นอกจากนี้ยังสามารถสร้าง TensorFlow โดยใช้เชลล์ MSYS ทำการเปลี่ยนแปลงตามรายการด้านล่าง จากนั้นทำตามคำแนะนำก่อนหน้าสำหรับบรรทัดคำสั่งดั้งเดิมของ Windows ( cmd.exe )

ปิดใช้งานการแปลงเส้นทาง MSYS

MSYS แปลงอาร์กิวเมนต์ที่มีลักษณะเหมือนพาธ Unix เป็นพาธ Windows โดยอัตโนมัติ และสิ่งนี้ใช้ไม่ได้กับ bazel (ป้ายกำกับ //path/to:bin ถือเป็นเส้นทางสัมบูรณ์ของ Unix เนื่องจากเริ่มต้นด้วยเครื่องหมายทับ)

export MSYS_NO_PATHCONV=1
export MSYS2_ARG_CONV_EXCL="*"

กำหนดเส้นทางของคุณ

เพิ่มไดเร็กทอรีการติดตั้ง Bazel และ Python ให้กับตัวแปรสภาพแวดล้อม $PATH ของคุณ หากติดตั้ง Bazel ไว้ที่ C:\tools\bazel.exe และ Python เป็น C:\Python\python.exe ให้ตั้งค่า PATH ของคุณด้วย:

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

สำหรับการรองรับ GPU ให้เพิ่มไดเร็กทอรี CUDA และ cuDNN ไปที่ $PATH :

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

การกำหนดค่าบิลด์ทดสอบแล้ว

ซีพียู

จีพียู