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

时间：2019-01-29 11:23:06 阅读：230 评论：0 收藏：0 [点我收藏+]

我们先创建一个用于caffe和tensorflow共存的虚拟环境,安装完成后激活环境.

conda create -n caffe_tf_36 python=3.6

source activate caffe_tf_36

技术分享图片

conda install -c anaconda caffe-gpu

技术分享图片

conda install protobuf=3.2

技术分享图片

conda install tensorflow-gpu

技术分享图片

测试:

import caffe
import tensorflow as tf

sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

技术分享图片

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

踩

(0)

评论一句话评论（0）

分享档案

更多>