作者:超级娱乐测试_499 | 来源:互联网 | 2023-02-10 16:32
1、创建虚拟环境 我们先创建一个用于caffe和tensorflow共存的虚拟环境,安装完成后激活环境.condacreate-ncaffe_tf_36python3.6sourc
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))