[Tools: cntk] Linux下cntk-gpu安装

CNTK官方安装教程：https://learn.microsoft.com/zh-cn/cognitive-toolkit/setup-linux-python?tabs=cntkpy27

1. 在conda中创建并激活python=3.6的新环境

conda create -n cntk python=3.6
conda activate cntk

2. 安装OpenMPI

yum install openmpi-bin

3. 安装cntk-gpu

pip install cntk-gpu

报错：

• ModuleNotFoundError: No module named 'cntk._cntk_py'
• ImportError: libmpi_cxx.so.1: cannot open shared object file: No such file or directory

问题原因：LD_LIBRARY_PATH路径下没有libmpi_cxx.so.1包导致的。

解决方法：

• 在conda的cntk环境下安装OpenMPI后，/path-to-anaconda3/envs/cntk/lib/路径下有libmpi_cxx.so.xx和libmpi.so.xx.10.1两个包，其中xx为对应版本。
• echo ${LD_LIBRARY_PATH}，确定LD_LIBRARY_PATH路径。
• 在LD_LIBRARY_PATH路径下，建立软连接：

ln -s /path-to-anaconda3/envs/cntk/lib/libmpi_cxx.so.20 ${LD_LIBRARY_PATH}/libmpi_cxx.so.1
ln -s /path-to-anaconda3/envs/cntk/lib/libmpi.so.20.10.1 ${LD_LIBRARY_PATH}/libmpi.so.12

• 测试是否成功

python -c "import cntk; print(cntk.__version__)"

• 在Pycharm中调试则需要将${LD_LIBRARY_PATH}添加到环境路径：

参考：

https://youtrack.jetbrains.com/issue/PY-29580/PyCharm-doesnt-set-LDLIBRARYPATH-when-running-scripts