Caffe和TensorFlow Protobuf - 保持不同的版本

时间:2015-12-09 21:44:36

标签: python version protocol-buffers caffe tensorflow

我已经使用protobuf 2.6.1在我的Ubuntu15.10x64计算机上安装了Caffe。此外,截至今天,我使用virtualenv安装了TensorFlow,但TensorFlow需要protobuf 3.0。根据我的阅读,virtualenv支持为python项目使用不同的库。我可以安全pip install --upgrade protobuf在(tensorflow)virtualenv中,而不会破坏我的Caffe安装 - 因此默认为protobuf 2.6,而TensorFlow使用的virtualenv中有protobuf 3.0吗?

或者是使用Docker安装或在虚拟机中创建克隆的唯一解决方案?总比没有好,但我想用我的GPU和全功率。

感谢您的帮助和澄清。

编辑:

所以,当我尝试更新protobuf时会发生这种情况 - >

(tensorflow)peter@UNIGMA:~$ pip install --upgrade protobuf 
Requirement already up-to-date: protobuf in /usr/local/lib/python2.7/dist-packages 
Requirement already up-to-date: setuptools in ./tensorflow/lib/python2.7/site-packages (from protobuf) 
Cleaning up... 

如果我检查版本,则返回 - >

(tensorflow)peter@UNIGMA:~$ pip show protobuf
---
Name: protobuf
Version: 2.6.1
Location: /usr/local/lib/python2.7/dist-packages
Requires: setuptools
(tensorflow)peter@UNIGMA:~$ pip -V
pip 1.5.6 from /home/peter/tensorflow/local/lib/python2.7/site-packages (python 2.7)

我是否必须手动更新protobuf?我假设我会使用我需要的版本在protobuf内创建一个./tensorflow/lib/python2.7/dist-packages文件夹?

1 个答案:

答案 0 :(得分:1)

请注意,你的virtualenv没有看到来自外部的软件包,所以你根本不应该在你的virtualenv中看到protobuf,即使你已经在系统中全局安装(或者在Caffe的virtualenv中,取决于你的设置)

在TensorFlow的virtualenv中运行pip install protobuf应该是安全的,它不会破坏你的全局设置或任何其他虚拟现实。