windows下安装配置TensorFlow,caffe的Python接口

由于TensorFlow在windows下只能支持Python3.5和Python3.6,所以在安装Python时需要安装Python3.5以上版本。推荐安装anaconda3。


一、安装TensorFlow官方有两种方法:

方法1:使用pip安装

在安装好Python环境后,打开cmd或者powershell,安装cpu-only版TensorFlow,执行

pip install --upgrade tensorflow

安装gpu版,则执行
pip install --upgrade tensorflow-gpu

方法2:使用anaconda安装

安装好anaconda后,打开Anaconda Prompt(开始->Anaconda3->Anaconda Prompt),执行

conda create -n tensorflow python=3.5

完成后激活TensorFlow环境

activate tensorflow

之后安装对应版本的TensorFlow

pip install --ignore-installed --upgrade tensorflow
pip install --ignore-installed --upgrade tensorflow-gpu


安装完成后打开Python,执行

import tensorflow as tf
没有报错,说明安装成功。


二、安装配置caffe

caffe可以通过编译安装,可以参考Caffe编译(Win10+Vs2015+Cuda8.0)

编译安装比较费时,如果不是学习目的,只是需要使用caffe,可以直接使用编译好的caffe配置接口。

从github上选择下载对应环境编译好的caffe,地址为https://github.com/BVLC/caffe/tree/windows

解压后将python目录下的caffe文件夹复制到C:\Users\chmar\Anaconda3\Lib\site-packages下即可。如果没有安装anaconda,将caffe复制到python目录下的\Lib\site-packages即可。

打开python,执行

import caffe
没有报错,安装成功。

若出现

ImportError: No module named google.protobuf.internal
则在cmd中执行

conda install protobuf
没安装anaconda,执行

pip install protobuf

你可能感兴趣的:(tensorflow,caffe)