1. 程式人生 > >Anaconda建立caffe和tensorflow共存環境

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即可。