你当前正在访问 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_PATH
CUDA_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 编译器文档。 |
详细信息