Anaconda創建caffe和tensorflow共存環境

一、前言

 

安裝環境：

Anaconda

Ubuntu

二、安裝步驟

我們分幾步進行，anconda的安裝和使用方法就不講解了。我們直接安裝caffe和tensorflow。

1、創建虛擬環境

我們先創建一個用於caffe和tensorflow共存的虛擬環境：

conda create -n caffe-tf python=3.6

直接回車，安裝即可。安裝好后，進入到caffe-tf虛擬環境：

source activate caffe-tf

2、安裝caffe

我們先安裝caffe，一定要先安裝caffe！

conda install caffe

安裝的時候，我們可以看到，默認安裝的protobuf的版本是3.4。

 

3、降低protobuf版本

經過一定時間的等待，caffe安裝完成。我們接下來，降低protobuf的版本為3.2。因為只有這樣，caffe和tensorflow才能不沖突。

conda install protobuf=3.2

4、安裝tensorflow

做好上面幾步之后，我們就可以直接安裝tensorflow-gpu了。

conda install tensorflow

這樣protobuf就不用變動了。

三、測試

安裝好這些之后，我們進行一下測試，看看是否都已經安裝成功了。

測試代碼如下：

import caffe
import tensorflow
 
print(dir(caffe))
print(dir(tensorflow))

 首先進入到caffe環境，再spyder即可。