בנה ממקור ב-Windows

בנה חבילת TensorFlow pip מהמקור והתקן אותה ב-Windows.

התקנה עבור Windows

התקן את כלי הבנייה הבאים כדי להגדיר את סביבת הפיתוח של Windows שלך.

התקן את Python ואת התלות בחבילת TensorFlow

התקן מהדורת Python 3.7+ 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, ראה את תצורות ה-build שנבדקו עבור Windows. הגדר את Bazel לבנות C++ .

הוסף את המיקום של קובץ ההפעלה של Bazel למשתנה הסביבה %PATH% שלך.

התקן את MSYS2

התקן את MSYS2 עבור כלי הפח הדרושים לבניית TensorFlow. אם MSYS2 מותקן ב- C:\msys64 , הוסף את C:\msys64\usr\bin למשתנה הסביבה %PATH% שלך. לאחר מכן, באמצעות cmd.exe , הרץ:

pacman -S git patch unzip

התקן את Visual C++ Build Tools 2019

התקן את כלי הבנייה של Visual C++ 2019 . זה מגיע עם Visual Studio 2019 אך ניתן להתקין אותו בנפרד:

1. עבור אל ההורדות של Visual Studio ,
2. בחר ניתנים להפצה מחדש וכלי בנייה ,
3. הורד והתקן:
   • Microsoft Visual C++ 2019 ניתן להפצה מחדש
   • Microsoft Build Tools 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

בניית TensorFlow מוגדרת על ידי קובץ .bazelrc בספריית השורש של המאגר. ניתן להשתמש בסקריפטים ./configure או ./configure.py כדי להתאים הגדרות נפוצות.

אם אתה צריך לשנות את התצורה, הפעל את הסקריפט ./configure מספריית השורש של המאגר.

python ./configure.py

סקריפט זה מבקש ממך את מיקום התלות של TensorFlow ומבקש אפשרויות תצורת בנייה נוספות (דגלים מהדר, למשל). להלן תצוגה לדוגמה של python ./configure.py (ההפעלה שלך עשויה להיות שונה):

python ./configure.py
Starting local Bazel server and connecting to it...
................
You have bazel 0.15.0 installed.
Please specify the location of python. [Default is C:\python36\python.exe]:

Found possible Python library paths:
  C:\python36\lib\site-packages
Please input the desired Python library path to use.  Default is [C:\python36\lib\site-packages]

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

Please specify the CUDA SDK version you want to use. [Leave empty to default to CUDA 9.0]:

Please specify the location where CUDA 9.0 toolkit is installed. Refer to README.md for more details. [Default is C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v9.0]:

Please specify the cuDNN version you want to use. [Leave empty to default to cuDNN 7.0]: 7.0

Please specify the location where cuDNN 7 library is installed. Refer to README.md for more details. [Default is C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v9.0]: C:\tools\cuda

Please specify a list of comma-separated Cuda compute capabilities you want to build with.
You can find the compute capability of your device at: https://developer.nvidia.com/cuda-gpus
Please note that each additional compute capability significantly increases your build time and binary size. [Default is: 3.5,7.0]: 3.7

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.

Configuration finished

אפשרויות תצורה

לתמיכה ב-GPU , ציין את הגרסאות של CUDA ו-cuDNN. אם למערכת שלך מותקנות מספר גרסאות של CUDA או cuDNN, הגדר במפורש את הגרסה במקום להסתמך על ברירת המחדל. ./configure.py יוצר קישורים סמליים לספריות ה-CUDA של המערכת שלך - כך שאם תעדכן את נתיבי ספריית ה-CUDA שלך, יש להפעיל שוב את שלב ההגדרה הזה לפני הבנייה.

בנה והתקן את חבילת ה-pip

חבילת ה-pip נבנית בשני שלבים. פקודות bazel build יוצרות תוכנית "בונה חבילות". לאחר מכן אתה מפעיל את בונה החבילות כדי ליצור את החבילה.

בנה את בונה החבילות

tensorflow:master repo עודכן ל-build 2.x כברירת מחדל. התקן את Bazel והשתמש ב- bazel build כדי ליצור את בונה החבילות של TensorFlow.

bazel build //tensorflow/tools/pip_package:build_pip_package

מעבד בלבד

השתמש bazel כדי להפוך את בונה החבילות של TensorFlow עם תמיכה במעבד בלבד:

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

השתמש באפשרות זו בעת בנייה כדי למנוע בעיות ביצירת חבילה: 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-cp36-cp36m-win_amd64.whl

בנה באמצעות מעטפת MSYS

ניתן לבנות את TensorFlow גם באמצעות מעטפת MSYS. בצע את השינויים המפורטים להלן, ולאחר מכן עקוב אחר ההוראות הקודמות עבור שורת הפקודה המקורית של Windows ( cmd.exe ).

השבת המרת נתיב MSYS

MSYS ממירה אוטומטית ארגומנטים שנראים כמו נתיבי Unix לנתיבי Windows, וזה לא עובד עם bazel . (התווית //path/to:bin נחשבת לנתיב אבסולוטי של יוניקס מכיוון שהיא מתחילה בקו נטוי.)

export MSYS_NO_PATHCONV=1
export MSYS2_ARG_CONV_EXCL="*"

הגדר את ה-PATH שלך

הוסף את ספריות ההתקנה של Bazel ו-Python למשתנה הסביבתי $PATH שלך. אם Bazel מותקן ב- C:\tools\bazel.exe , ו-Python ל- C:\Python36\python.exe , הגדר את ה- PATH שלך עם:

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

לתמיכה ב-GPU, הוסף את ספריות ה-CUDA וה-cuDNN bin ל- $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"

תצורות בנייה נבדקו

מעבד

GPU