你当前正在访问 Microsoft Azure Global Edition 技术文档网站。 如果需要访问由世纪互联运营的 Microsoft Azure 中国技术文档网站,请访问 https://docs.azure.cn

Windows环境变量

如果要在自己的计算机上生成Microsoft Cognitive Toolkit,则需要安装多个外部组件。 生成过程通过环境变量在计算机上查找这些已安装的组件。 这样,你可以根据自己的愿望和首选项安装这些组件。

此页列出了CNTK生成过程使用的环境变量。 它还列出了这些组件 的首选 位置。 首选位置是镜像内部自动生成和测试计算机的配置。 首选位置也是文档中用于描述安装过程的位置。 如前所述,这只是 首选 位置,你可以将其调整为你的喜好。

环境变量 首选位置
BOOST_INCLUDE_PATH c:\local\boost_1_60_0-msvc-14.0 所需的 BOOST ()
BOOST_LIB_PATH c:\local\boost_1_60_0-msvc-14.0\lib64-msvc-14.0 所需的 BOOST ()
CNTK_OPENBLAS_PATH c:\local\CNTKopenBLAS 适用于 CNTK UWP 生成 (可选) 的 OpenBLAS 库
CUB_PATH c:\local\cub-1.7.4 GPU 生成) 所需的 NVIDIA CUB 库 (
CUDNN_PATH C:\local\cudnn-9.0-v7.0\cuda GPU 生成) 所需的 NVIDIA cuDNN (
CUDA_PATH C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0 GPU 生成) 所需的 NVIDIA CUDA (
CUDA_PATH_V9_0 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0 GPU 生成) 所需的 NVIDIA CUDA (
MKL_PATH C:\local\mklove my life-mkldnn-0.14 需要) MKL_PATH (
OPENCV_PATH_V31 c:\local\Opencv3.1.0\build ImagerReader 生成和 TensorBoard 的映像功能) 所需的 OpenCV (
PROTOBUF_PATH c:\local\protobuf-3.1.0-vs17 需要 protobuf 库 ()
SWIG_PATH C:\local\swigwin-3.0.10 CNTK Python、Java 和 C#/.NET 支持) 所需的 SWIG (
ZLIB_PATH c:\local\zlib-vs17 ImagerReader 生成) 所需的压缩库 (
VS2017INSTALLDIR C:\Program Files (x86) \Microsoft Visual Studio\version\offering 使用 VS2017) 生成 CNTK 2.4+ 所需的Windows (CNTK

在命令行界面中,可以使用以下命令在计算机上设置环境变量:

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

CUDA_PATHCUDA_PATH_V9_0不应直接更改环境变量。 NVIDIA CUDA 安装程序直接定义这些变量。

有几个其他环境变量可用于定义在系统上生成的CNTK功能。 如果没有在系统上设置这些环境变量,则假定默认值。

环境变量 默认值
CNTK_ENABLE_ASGD true 数据并行 ASGD
CNTK_ENABLE_1BITSGD false 1 位 SG
CNTK_PY27_PATH CNTK Python 2.7 环境的路径
CNTK_PY35_PATH CNTK Python 3.5 环境的路径
CNTK_PY36_PATH CNTK Python 3.6 环境的路径

假设默认值:

  • 正在生成 DataParallel ASGD
  • 未生成 1 位 SG
  • 未定义 CNTK_PY.._PATH时, 生成 Python 库

其他环境变量

还有其他环境变量可能会影响编译过程:

环境变量
CNTK_CUDA_CODEGEN_DEBUG CNTK_CUDA_CODEGEN_RELEASE 使用这些环境变量,可以定义 NVidia 编译器目标体系结构。 例如,将变量设置为 compute_35,sm_35;compute_50,sm_50 仅生成级别 3.5 和 5.0 兼容的 cubin 和 PTX 信息。 有关此文档的详细信息,请参阅 NVidia 编译器文档。

详细信息