[Tools: cntk] Linux下cntk-gpu安装
创始人
2024-03-20 23:17:45

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

相关内容

热门资讯

超有范的个性伤感说说 超级经典... 超有范的个性伤感说说  1、当爱己成往事时,是否还能记住彼此。  2、很多事都介于不说憋屈和说了矫情...
给你心灵以慰藉的爱情说说 触动... 给你心灵以慰藉的爱情说说  1、谢谢你,让我的寂寞有了寄托。  2、我不怕离别和伤害我只是怕空荡的心...
科森科技股价涨5.09%,红土... 2月13日,科森科技涨5.09%,截至发稿,报28.69元/股,成交4.08亿元,换手率2.63%,...
空间个性非主流伤感说说 非主流... 空间个性非主流伤感说说  1、我爱你,是我的事,与你无关。  2、围绕在你身边的人太多,阻挡了我爱你...
句句都入木三分的经典说说 句句... 句句都入木三分的经典说说  1、一个人炫耀什么,说明内心缺少什么。  2、失言就是一不小心说了实话。...