1、创建虚拟环境

 我们先创建一个用于caffe和tensorflow共存的虚拟环境,安装完成后激活环境.

conda create -n caffe_tf_36 python=3.6

source activate caffe_tf_36

基于Anaconda安装tensorflow-GPU和caffe-GPU

2. 安装caffe-gpu

conda install -c anaconda caffe-gpu

基于Anaconda安装tensorflow-GPU和caffe-GPU 

3. 降低protobuf版本

conda install protobuf=3.2

基于Anaconda安装tensorflow-GPU和caffe-GPU

4、安装tensorflow-gpu

conda install tensorflow-gpu

 

基于Anaconda安装tensorflow-GPU和caffe-GPU

 

 


测试:

import caffe
import tensorflow as tf

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

 

基于Anaconda安装tensorflow-GPU和caffe-GPU