Compartir a través de


variables de entorno de Windows

Si va a compilar el Microsoft Cognitive Toolkit en su propia máquina, requiere la instalación de varios componentes externos. El proceso de compilación busca estos componentes instalados en la máquina a través de variables de entorno. Esto le permite instalar estos componentes según sus preferencias y deseos.

En esta página se enumeran las variables de entorno que usa el proceso de compilación de CNTK. También enumera la ubicación preferida para estos componentes. La ubicación preferida es reflejar la configuración de nuestras máquinas internas de compilación y prueba automatizadas. La ubicación preferida también es la ubicación que se usa en la documentación para describir el proceso de instalación. Como se dijo antes, esta es solo la ubicación preferida y usted puede ajustarla a su gusto.

Variable de entorno Ubicación preferida
BOOST_INCLUDE_PATH c:\local\boost_1_60_0-msvc-14.0 BOOST (obligatorio)
BOOST_LIB_PATH c:\local\boost_1_60_0-msvc-14.0\lib64-msvc-14.0 BOOST (obligatorio)
CNTK_OPENBLAS_PATH c:\local\CNTKopenBLAS Biblioteca OpenBLAS para CNTK compilación para UWP (opcional)
CUB_PATH c:\local\cub-1.7.4 Biblioteca DE NVIDIA CUB (necesaria para la compilación de GPU)
CUDNN_PATH C:\local\cudnn-9.0-v7.0\cuda NVIDIA cuDNN ( necesario para la compilación de GPU)
CUDA_PATH C:\Archivos de programa\NVIDIA GPU Computing Toolkit\CUDA\v9.0 NVIDIA CUDA (necesario para la compilación de GPU)
CUDA_PATH_V9_0 C:\Archivos de programa\NVIDIA GPU Computing Toolkit\CUDA\v9.0 NVIDIA CUDA (necesario para la compilación de GPU)
MKL_PATH C:\local\mklove my life-mkldnn-0.14 MKL_PATH (obligatorio)
OPENCV_PATH_V31 c:\local\Opencv3.1.0\build OpenCV (necesario para la compilación imagerReader y la característica image de TensorBoard)
PROTOBUF_PATH c:\local\protobuf-3.1.0-vs17 Biblioteca Protobuf (obligatorio)
SWIG_PATH C:\local\swigwin-3.0.10 SWIG (necesario para CNTK compatibilidad con Python, Java y C#/.NET)
ZLIB_PATH c:\local\zlib-vs17 Biblioteca de compresión (necesaria para la compilación imagerReader)
VS2017INSTALLDIR C:\Archivos de programa (x86)\Microsoft Visual Studio\version\offering CNTK en Windows (necesario para compilar CNTK 2.4+ con VS2017)

En un shell de comandos, puede usar estos comandos para establecer las variables de entorno en el equipo:

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

Las CUDA_PATH variables de entorno y CUDA_PATH_V9_0 no deben cambiarse directamente. El instalador de NVIDIA CUDA define estas variables directamente.

Hay varias variables de entorno adicionales que se pueden usar para definir las características de CNTK que se basan en el sistema. Si no tiene estas variables de entorno establecidas en el sistema, se asume el valor predeterminado.

Variable de entorno Valor predeterminado
CNTK_ENABLE_ASGD true ASGD paralelo de datos
CNTK_ENABLE_1BITSGD false SGD de 1bit
CNTK_PY27_PATH Ruta de acceso al entorno de Python 2.7 de CNTK
CNTK_PY35_PATH Ruta de acceso al entorno de Python 3.5 de CNTK
CNTK_PY36_PATH Ruta de acceso al entorno de Python 3.6 de CNTK

Suponiendo que los valores predeterminados:

  • Se está compilando DataParallel ASGD
  • SgD de 1bit no se está compilando
  • Con undefined CNTK_PY.._PATH, las bibliotecas de Python no se están compilando

Variables de entorno adicionales

Hay variables de entorno adicionales que pueden influir en el proceso de compilación:

Variables de entorno
CNTK_CUDA_CODEGEN_DEBUG CNTK_CUDA_CODEGEN_RELEASE Con estas variables de entorno puede definir las arquitecturas de destino del compilador de NVidia. Por ejemplo, si se establece una variable en compute_35,sm_35;compute_50,sm_50 , solo se compilará el nivel 3.5 y 5.0 compatible con la información de cubin y PTX. Para obtener información detallada sobre esto, consulte la documentación del compilador de NVidia.

Más información