在C ++中嵌入python时如何修复“ ImportError:没有名为'tensorflow'的模块”

时间:2019-03-31 08:16:10

标签: c++ python-3.x tensorflow embedding

我的问题是关于在C ++中嵌入python。我想嵌入一个导入tensorflow和numpy的python模块。我能够嵌入一个无法成功导入tensorflow或numpy的python代码。另外,在我的python中使用tensorflow也没有问题。 我的设置是 操作系统:Linux Ubuntu 16.04 Python版本:3.5

我使用PyRun_SimpleString()测试了各种情况。我尝试了Importing tensorflow when embedding python in c++ returns null

Py_Initialize();
PySys_SetArgv(argc, (wchar_t**)argv);
PyRun_SimpleString("import os \n"
                       "print('Hello TF!!!')");)
Py_Finalize();

但是它给出了错误

Fatal Python error: no mem for sys.argv
ValueError: character U+65442f2e is not in range [U+0000; U+10ffff]

Current thread 0x00007f5a69506740 (most recent call first):
Aborted (core dumped)

这是我的main.cpp和CMakeLists.txt文件:

main.cpp:

#include "Python.h"
#include <iostream>

using namespace std;

int main(int argc, char *argv[]) {

    Py_Initialize();
    PyRun_SimpleString("import tensorflow \n"
                       "print('Hello TF!!!')");
    Py_Finalize();
    return 0;
}

CMakeLists.txt

cmake_minimum_required(VERSION 3.13)
project(Demo)

set(CMAKE_CXX_STANDARD 14)
SET(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -I/usr/include/python3.5 -I/usr/include/x86_64-linux-gnu/python3.5  -fno-strict-aliasing -Wdate-time -D_FORTIFY_SOURCE=2 -g -fstack-protector-strong -Wformat -Werror=format-security  -DNDEBUG -g -fwrapv -O2 -Wall -Wstrict-prototypes")

add_executable(Demo main.cpp)

set(PYTHON_EXECUTABLE "/usr/bin/python3.5")
set(PYTHON_INCLUDE_DIR "/usr/include/python3.5m")
set(PYTHON_LIBRARIES "/usr/lib/x86_64-linux-gnu/libpython3.5m.so")

include_directories(${PYTHON_INCLUDE_DIRS})
include_directories(${PYTHON_DIRECTORIES})
target_link_libraries(Demo ${PYTHON_LIBRARIES})

使用命令构建代码:

cmake --build . --target Demo -- -j 2

使用命令执行:

./Demo

执行程序时,出现以下错误提示:

Traceback (most recent call last):
  File "<string>", line 1, in <module>
ImportError: No module named 'tensorflow'

如果我使用“ import os” /“ import time”代替“ import tensorflow”,则使用“ Hello TF !!!”。是结果。

如何解决此错误? CMakeList.txt中是否缺少任何内容?

1 个答案:

答案 0 :(得分:0)

我又按照https://www.tensorflow.org/install/pip中给出的步骤安装了Tensorflow,并在创建的虚拟环境中执行了上述代码。然后,问题解决了。