海思AI芯片(HI35xx):tensorflow转caffemodel之环境和权限问题

在pycharm中直接运行tf113tocaffe.py报错如下:
cudnn_conv_layer.cpp:52] Check failed: error == cudaSuccess (30 vs. 0) unknown error
如下图所示:
海思AI芯片(HI35xx):tensorflow转caffemodel之环境和权限问题
解决方法:
在tf113tocaffe.py所在目录下,打开终端:

sudo python3 tf113tocaffe.py    (重点是sudo python3)

问题2:
在终端运行:sudo python3 tf113tocaffe.py
报错如下: ImportError: No module named “tensorflow”, 如下图所示:
海思AI芯片(HI35xx):tensorflow转caffemodel之环境和权限问题
解决方法:
由于sudo python3环境下,没有安装tensorflow,所以终端输入命令如下:

sudo pip3 install --ignore-installed six tensorflow==1.13.1 -i https://pypi.doubanio.com/simple

输入此命令报错如下:
ERROR: Command errored out with exit status 1: python setup.py egg_info Check the logs for full command output.
如下图所示:
海思AI芯片(HI35xx):tensorflow转caffemodel之环境和权限问题
解决方法:

sudo pip3 install --upgrade setuptools

然后再安装tensorflow
即:

sudo pip3 install --ignore-installed six tensorflow==1.13.1 -i https://pypi.doubanio.com/simple

此次可以顺利安装

再次在tf113tocaffe.py所在目录下,打开终端:

sudo python3 tf113tocaffe.py    (重点是sudo python3)

测试tensorflow与caffe都没有问题,即Check failed: error == cudaSuccess (30 vs. 0) unknown error问题解决,即caffe与tensoflow1.13.1并存,可以进行模型转换。

上一篇:海思AI芯片(HI35xx):tensorflow转caffemodel之numpy版本问题


下一篇:补充