Partager via

Variables d’environnement Windows

  • Article

Si vous générez le Microsoft Cognitive Toolkit sur votre propre ordinateur, il nécessite l’installation de plusieurs composants externes. Le processus de génération recherche ces composants installés sur votre ordinateur via des variables d’environnement. Cela vous permet d’installer ces composants en fonction de vos souhaits et préférences.

Cette page répertorie les variables d’environnement utilisées par le processus de génération CNTK. Il répertorie également l’emplacement préféré de ces composants. L’emplacement préféré reflète la configuration de nos machines de build et de test automatisées internes. L’emplacement préféré est également l’emplacement utilisé dans la documentation pour décrire le processus d’installation. Comme nous l’avons dit précédemment, il s’agit uniquement de l’emplacement préféré et vous pouvez l’ajuster à votre goût.

Variable d’environnement Emplacement préféré
BOOST_INCLUDE_PATH c:\local\boost_1_60_0-msvc-14.0 BOOST (obligatoire)
BOOST_LIB_PATH c:\local\boost_1_60_0-msvc-14.0\lib64-msvc-14.0 BOOST (obligatoire)
CNTK_OPENBLAS_PATH c:\local\CNTKopenBLAS Bibliothèque OpenBLAS pour CNTK build UWP (facultatif)
CUB_PATH c:\local\cub-1.7.4 Bibliothèque NVIDIA CUB (requise pour la build GPU)
CUDNN_PATH C:\local\cudnn-9.0-v7.0\cuda NVIDIA cuDNN (requis pour la build GPU)
CUDA_PATH C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0 NVIDIA CUDA (requis pour la build GPU)
CUDA_PATH_V9_0 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0 NVIDIA CUDA (requis pour la build GPU)
MKL_PATH C:\local\mklove my life-mkldnn-0.14 MKL_PATH (obligatoire)
OPENCV_PATH_V31 c:\local\Opencv3.1.0\build OpenCV (requis pour la build ImagerReader et la fonctionnalité Image de TensorBoard)
PROTOBUF_PATH c:\local\protobuf-3.1.0-vs17 Bibliothèque Protobuf (obligatoire)
SWIG_PATH C:\local\swigwin-3.0.10 SWIG (requis pour CNTK prise en charge de Python, Java et C#/.NET)
ZLIB_PATH c:\local\zlib-vs17 Bibliothèque de compression (obligatoire pour la build ImagerReader)
VS2017INSTALLDIR C:\Program Files (x86)\Microsoft Visual Studio\version\offering CNTK sur Windows (obligatoire pour la génération de CNTK 2.4+ avec VS2017)

Dans un interpréteur de commandes, vous pouvez utiliser ces commandes pour définir les variables d’environnement sur votre ordinateur :

setx  BOOST_INCLUDE_PATH c:\local\boost_1_60_0-msvc-14.0
setx  BOOST_LIB_PATH c:\local\boost_1_60_0-msvc-14.0\lib64-msvc-14.0
setx  CUB_PATH c:\local\cub-1.7.4
setx  CUDNN_PATH C:\local\cudnn-9.0-v7.0\cuda
setx  MKL_PATH C:\local\mklml-mkldnn-0.14
setx  OPENCV_PATH_V31 c:\local\Opencv3.1.0\build
setx  PROTOBUF_PATH c:\local\protobuf-3.1.0-vs17
setx  SWIG_PATH c:\local\swigwin-3.0.10
setx  ZLIB_PATH c:\local\zlib-vs17

Les CUDA_PATH variables d’environnement ne CUDA_PATH_V9_0 doivent pas être modifiées directement par vous. Le programme d’installation NVIDIA CUDA définit directement ces variables.

Il existe plusieurs variables d’environnement supplémentaires qui peuvent être utilisées pour définir les fonctionnalités de CNTK que vous générez sur votre système. Si ces variables d’environnement ne sont pas définies sur votre système, la valeur par défaut est supposée.

Variable d’environnement Valeur par défaut
CNTK_ENABLE_ASGD true ASGD parallèle de données
CNTK_ENABLE_1BITSGD false SGD 1bit
CNTK_PY27_PATH Chemin d’accès à l’environnement Python 2.7 CNTK
CNTK_PY35_PATH Chemin d’accès à l’environnement Python 3.5 CNTK
CNTK_PY36_PATH Chemin d’accès à l’environnement Python 3.6 CNTK

En supposant les valeurs par défaut :

  • DataParallel ASGD est généré
  • SGD 1bit n’est pas généré
  • Avec des bibliothèques Python non définies CNTK_PY.._PATH, elles ne sont pas générées

Variables d’environnement supplémentaires

Il existe des variables d’environnement supplémentaires qui peuvent influencer le processus de compilation :

Variables d'environnement
CNTK_CUDA_CODEGEN_DEBUG CNTK_CUDA_CODEGEN_RELEASE Avec ces variables d’environnement, vous pouvez définir les architectures cibles du compilateur NVidia. Par exemple, la définition d’une variable pour compute_35,sm_35;compute_50,sm_50 générera uniquement les informations de niveau 3.5 et 5.0 compatibles cubin et PTX. Pour plus d’informations sur ce sujet, reportez-vous à la documentation du compilateur NVidia.

Plus d’informations