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))
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:基于Anaconda安装tensorflow-GPU和caffe-GPU - Python技术站