Compilation personnalisée de Tensorflow sous Windows

Question

Donc, j'ai installé Bazel via Chocolatey, installé Python 3.5 et 2.7, installé CUDA v8, et cuDNN v6, et installé JDK 8.0, j'essaie maintenant de construire sur mesure TensorFlow sur mon appareil Windows 10, avec AVX, AVX 2 et CUDA. TensorFlow-GPU, la version préconstruite, fonctionne, je l'ai déjà testée et exécutée avec succès.

J'ai suivi les instructions d'autres articles, à la fois sur le site actuel de TensorFlows (en essayant d'adapter certaines sections des installations Linux/Mac), et ici. Le plus loin que j'ai fait est ; cloner le dépôt Github via Msys2, exécuter configure.py, puis essayer de construire via bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package Je reçois une erreur dont l'en-tête est :

Erreur de lecture de java.io.IOException : CreateProcess() : Le système ne trouve pas le fichier spécifié. : C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v8.0/include/cudnn.h

J'ai vérifié deux fois, ce fichier existe bien, donc je ne sais pas pourquoi j'obtiens cette erreur.

EDIT : J'ai également essayé de l'exécuter via Powershell, mais j'ai obtenu le même résultat.

Toute aide serait très appréciée.

Answer 1

J'ai eu exactement la même erreur en essayant de construire Tensorflow sous Windows (en utilisant cuDNN 5.1). J'ai résolu le problème en lançant bazel depuis le terminal msys2 (au lieu de l'invite de commande Windows) et en définissant manuellement la variable d'environnement BAZEL_SH avant de tenter de construire.

export BAZEL_SH=c:/tools/msys64/usr/bin/bash.exe
bazel build -c opt --config=win-cuda tensorflow/cc:cc_ops

Answer 2

Les étapes suivantes m'ont aidé à compiler Tensorflow sur Windows 10.

pacman -Syuu patch
ln -s "c:\python27\python.exe" /usr/bin/python

export BAZEL_SH=c:/tools/msys64/usr/bin/bash.exe
"C:\Documents and Settings\All Users\chocolatey\bin\bazel.exe" build --config=opt --config=win-cuda //tensorflow/tools/pip_package:build_pip_package

Mais après une heure de compilation, j'ai eu une autre erreur :

C:\tools\msys64\tmp_bazel_dmitry\x1e5egqw\execroot\org_tensorflow\external\protobuf_archive\python\google\protobuf\internal\api_implementation.cc : erreur fatale C1083 : Impossible d'ouvrir le fichier généré par le compilateur : '' : Invalide argument Target //tensorflow/tools/pip_package:build_pip_package n'a pas réussi à construire