这几天需要安装caffe,但是大部分教程都没有说怎么在已经有tensorflow的同时安装caffe。在这里记录一下,主要写一下整个安装过程中的大致流程,忽略详细步骤。
首先需要安装anaconda,这里安装anaconda的主要原因是,它能够管理环境,也就是说可以将tensorflow和caffe的安装环境隔离开。(网上有不少文章已经表示,直接将两者装在一个环境下会出问题)。anaconda的版本无所谓,3.6也没关系,因为在里面可以新建环境。
假设你已经安装好了anaconda,打开anaconda Navigator,创建两个python3.5新的环境,分别命名为caffe和tensorflow。完成后,如下图所示:
tensorflow的安装
1、首先打开anaconda prompt,输入activate tensorflow,这样表面你后续的安装操作,都是在tensorflow这个环境下进行的。所有的安装依赖都可以在以下目录中查询。
E:\studySoftware\anaconda\envs\tensorflow\Lib\site-packages
2、安装GPU版本需要CUDA8.0和cudnn5.1(这里一定要安装cudnn5.1版本,这是因为caffe只能支持这个版本)
CUDA安装:https://developer.nvidia.com/cuda-downloads
CuDNN安装:https://developer.nvidia.com/cudnn
3、安装numpy
pip install numpy
由于网络的某些原因,可能会导致通过pip安装会很慢,可以选择直接去https://www.lfd.uci.edu/~gohlke/pythonlibs/#xgboost该网址用ctrl+F找到并下载对应的wheel文件,以 pip install numpy‑1.11.2+mkl‑cp35‑cp35m‑win_amd64.whl 的形式来安装。
4、安装tensorflow
pip install tensorflow-gpu
如果慢一样可以下载对应的wheel文件,输入pip install tensorflow_gpu-0.12.0rc0-cp35-cp35m-win_amd64.whl
所有的操作要在activate tensorflow 后进行,如图示例:(这里还有一大堆依赖包,不一一列举,pip能自动安装)
5、测试tensorflow
进入python,输入import tensorflow as tf,若这句话通过,基本上就可以表明tensorflow安装成功
安装caffe
1、首先打开anaconda prompt,一样输入activate caffe,所有的安装依赖一样也都可以在以下目录中查询。
E:\studySoftware\anaconda\envs\caffe\Lib\site-packages
2、上面已经安装过CUDA8.0和cudnn5.1了,所以这边可以跳过。
3、安装visual stduio 2015,我这边用的是2015版,官网上说2013也可以
4、安装cmake,最好下一个高版本,我一开始下了一个cmake3.4.1,虽然满足官网的要