Dies ist das Verfahren zum Erstellen von TensorFlow 2.2.0-rc0 (GPU-Version) unter Windows 10. Da detaillierte Erklärungen weggelassen werden und nur die Hauptpunkte beschrieben werden, finden Sie für einen etwas detaillierteren Inhalt Blog von dev.infohub.cc usw. Ich hoffe, Sie können sich darauf beziehen.
Verwenden Sie zum Erstellen die folgende Umgebung. CUDA, cuDNN, Python usw. sollten im Voraus in den Pfad eingegeben werden.
C: \ msys64 \ usr \ bin.bazel herunter. Benennen Sie es in exe um und speichern Sie es in einem Ordner mit einem Pfad. Starten Sie C: \ msys64 \ msys2_shell.cmd und fügen Sie die für den Build erforderlichen Pakete hinzu.
#Installation der erforderlichen Pakete
pacman -S git patch unzip
#Danach wird die Konsole von MSYS2 nicht mehr verwendet und endet.
exit
Wir werden TensorFlow unter Verwendung der virtuellen Python-Umgebung g: \ venvs \ build_tf2 zum Erstellen erstellen.
Bitte lesen Sie den Pfad usw. für jede Umgebung.
Verwenden Sie für die nachfolgenden Vorgänge den Konsolenbildschirm, der über die Eingabeaufforderung "x64 Native Tools for VS 2019" in Visual Studio 2019 gestartet wird.
#Erstellen und aktivieren Sie eine virtuelle Umgebung
python -m venv g:\venvs\build_tf2
cd /d g:\venvs\build_tf2
g:\venvs\build_tf2\Scripts\activate.bat
#Pip Upgrade (optional)
python -m pip install --upgrade pip
#Stellen Sie die für die Erstellung erforderlichen Pakete vor
pip install six numpy wheel
pip install keras_applications==1.0.8 --no-deps
pip install keras_preprocessing==1.1.0 --no-deps
# TensorFlow v2.2.0-Holen Sie sich den Quellcode für rc0
git clone https://github.com/tensorflow/tensorflow.git
cd tensorflow
git checkout v2.2.0-rc0
#Nicht benötigte Umgebungsvariablen löschen (optional)
#Alle Umgebungsvariablen werden als Parameter übergeben, wenn die Parameterzeichenfolge 32768 Zeichen überschreitet
# FATAL: Command line too long (34052 > 32768)Ein solcher Fehler tritt auf.
#Um dies zu verhindern, löschen Sie hier vorübergehend unnötige Umgebungsvariablen.
# _OLD_VIRTUAL_PATH hat in der Regel eine große Anzahl von Zeichen, daher ist es die erste Wahl, die gelöscht werden muss.
set _OLD_VIRTUAL_PATH=
#Konfigurationseinstellungen erstellen
#----------------------------------------------------------------
# <Einstellungsbeispiel>
# Location of Python : (Default)
# ROCm support : (Default = N)
# CUDA support : Y
# CUDA compute capabilities : 7.5 (Für Details https://developer.nvidia.com/cuda-Siehe gpus)
# Optimization flag : /arch:AVX2
# Override eigen... : (Default = Y)
#----------------------------------------------------------------
python ./configure.py
#Build (für CUDA)
#Es wird einige zehn Minuten bis Stunden dauern, bis der Build abgeschlossen ist
bazel build --config=opt --config=cuda --define=no_tensorflow_py_deps=true --copt=-nvcc_options=disable-warnings //tensorflow/tools/pip_package:build_pip_package
#Radpaket erstellen
# g:\tensorflow_Erstellen Sie ein Paket im Ordner pkg (dauert einige Minuten)
bazel-bin\tensorflow\tools\pip_package\build_pip_package g:\tensorflow_pkg
#Damit ist der TensorFlow-Build abgeschlossen. Beenden Sie mit dem Verlassen.
#----------------------------------------------------------------
# <Referenz>
#Bazels Arbeitsakte%UserProfile%_bazel_%UserName%Weil es in erstellt wird
#Wenn Sie es nicht benötigen, können Sie diesen Ordner löschen (Kapazität ca. 20 GB).
#----------------------------------------------------------------
exit
So stellen Sie den erstellten TensorFlow in einer neuen virtuellen Python-Umgebung bereit.
Hier ist ein Beispiel für das Erstellen einer neuen virtuellen Umgebung in g: \ venvs \ tf2.
#Erstellen und aktivieren Sie eine virtuelle Umgebung
python -m venv g:\venvs\tf2
g:\venvs\tf2\Scripts\activate.bat
#Pip Update (optional)
python -m pip install --upgrade pip
#Installieren Sie den erstellten TensorFlow (geben Sie einfach die erstellte Raddatei an).
pip install g:\tensorflow_pkg\tensorflow-2.2.0rc0-cp38-cp38-win_amd64.whl
#Funktionsprüfung
python -c "import tensorflow as tf; print(tf.__version__); print(tf.keras.__version__)"
python -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"
Ich habe den Eindruck, dass das Erstellen von TensorFlow 2.2.0-rc0 viel einfacher ist als in TensorFlow v1. Selbst in diesem Build bin ich beim Versuch von "python. / Configure.py" auf den Fehler "FATAL: Befehlszeile zu lang (34052> 32768)" gestoßen, und nichts anderes ist passiert. tat.
In vielen Fällen kann die DLL zur Laufzeit nicht gefunden werden und es tritt ein Fehler auf. Überprüfen Sie in diesem Fall, ob der Pfad zur erforderlichen Bibliothek wie CUDA vorhanden ist.
Recommended Posts