基於Anaconda安裝tensorflow-GPU和caffe-GPU

1、創建虛擬環境

 我們先創建一個用於caffe和tensorflow共存的虛擬環境,安裝完成后激活環境.

conda create -n caffe_tf_36 python=3.6

source activate caffe_tf_36

2. 安裝caffe-gpu

conda install -c anaconda caffe-gpu

 

3. 降低protobuf版本

conda install protobuf=3.2

4、安裝tensorflow-gpu

conda install tensorflow-gpu

 

 

 

---

測試:

import caffe
import tensorflow as tf

sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))