一、前言
安装环境:
Anaconda
Ubuntu
二、安装步骤
我们分几步进行,anconda的安装和使用方法就不讲解了。我们直接安装caffe和tensorflow。
1、创建虚拟环境
我们先创建一个用于caffe和tensorflow共存的虚拟环境:
conda create -n caffe-tf python=3.6
直接回车,安装即可。安装好后,进入到caffe-tf虚拟环境:
source activate caffe-tf
2、安装caffe
我们先安装caffe,一定要先安装caffe!
conda install caffe
安装的时候,我们可以看到,默认安装的protobuf的版本是3.4。
3、降低protobuf版本
经过一定时间的等待,caffe安装完成。我们接下来,降低protobuf的版本为3.2。因为只有这样,caffe和tensorflow才能不冲突。
conda install protobuf=3.2
4、安装tensorflow
做好上面几步之后,我们就可以直接安装tensorflow-gpu了。
conda install tensorflow
这样protobuf就不用变动了。
三、测试
安装好这些之后,我们进行一下测试,看看是否都已经安装成功了。
测试代码如下:
import caffe import tensorflow print(dir(caffe)) print(dir(tensorflow))
首先进入到caffe环境,再spyder即可。