码迷,mamicode.com
首页 > 其他好文 > 详细

基于Anaconda安装tensorflow-GPU和caffe-GPU

时间:2019-01-29 11:58:10      阅读:355      评论:0      收藏:0      [点我收藏+]

标签:ace   测试   activate   imp   div   bsp   python   span   anaconda   

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

标签:ace   测试   activate   imp   div   bsp   python   span   anaconda   

原文地址:https://www.cnblogs.com/geoffreyone/p/10333045.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!