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

Posted 2021-02-06 geoffreyone

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))