(转)用着anaconda,从caffe转到tensorflow遇到的各种问题

原文地址:https://www.cnblogs.com/yzkk/p/6039141.html

由于要用pip装支持gpu的tensorflow时,需要cuda-8.0,而我只是在服务器的一个虚拟机里工作,对于驱动之类的东西都不太敢碰,所以只能分开安装

在官网下回了cuda_8.0.44_linux.run之后,运行sh cuda_8.0.44_linux.run -- noexec -- target cuda8.0,这样就会不运行,只把.run文件解压了

进入到cuda8.0文件夹里的run_files,可以看到有3个run文件,运行cuda-linux64-rel-8.0.44-21122537.run就能装cuda-8.0而不更改驱动了

 

装好之后就装了tensorflow,然后又从github下找来了caffe-tensorflow来把caffe训练出的caffemodel转换成tensorflow可以用的

怎么转换在github的readme有说:

$ ./convert.py examples/mnist/lenet.prototxt --code-output-path=mynet.py

$ ./convert.py examples/mnist/lenet.prototxt --caffemodel examples/mnist/lenet_iter_10000.caffemodel --data-output-path=mynet.npy

 

之后想起cuda更新了,要把caffe重新编译一次,编译完之后,我的pycaffe就出问题了。。。

问题不知道什么时候做了什么导致的。。。。问题提示是

Intel MKL FATAL ERROR: Cannot load libmkl_avx2.so or libmkl_def.so

确认过LD_LIBRARY_PATH下绝对有包含这两个东西。。。

 

在安装tensorflow的时候我好像更新了anaconda的几个库,mkl可能在那时候导致的?

先解决问题再说。。上网搜了下,相关的东西不多。。还好发现有人说preload就可以了,试了一下,真可以:

$ export LD_PRELOAD=libmkl_rt.so

 

之后继续折腾tensorflow去用转化出来的代码和网络,结果抱错

ImportError: No module named kaffe.tensorflow

没有那就去import,给PYTHONPATH加上

1

/your_path/caffe-tensorflow

 就可以了

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值