docker下使用caffe的命令记录

Posted zhiyuanspring

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了docker下使用caffe的命令记录相关的知识,希望对你有一定的参考价值。

  • 查看所有的images
sudo docker images

技术分享

  • 利用某个image生成container
    sudo docker run -it --net=host -v /home/tingting/Dropbox/code/craterDetection/docker:/root/workspace tingting/caffe:version1 bash

技术分享

  • 打开jupyter notebook

jupyter notebook

技术分享

在浏览器中输入:localhost:8890

这里一定要注意,有时因为打开多个containers,新的container的jupyter notebook的接口是不同的,之前因为以为所有的接口窦唯8888,所以直接输入了8888,但其实这样打开的是8888那个接口对应的container

  • 将prototxt文件生成网络结构图
# Load the module
import caffe
import caffe.draw
from caffe.proto import caffe_pb2
from google.protobuf import text_format

# Set the parameters
input_net_proto_file = test_net.prototxt# net configuration file
output_image_file =test_net.jpg# the output image
rankdir =LR# net arrange method: LR、TB、RL

# Read the net
net = caffe_pb2.NetParameter()
text_format.Merge(open(input_net_proto_file).read(), net)

# Draw the net
print(Drawing net to %s‘ % output_image_file)
caffe.draw.draw_net_to_file(net, output_image_file, rankdir)
print(done...‘) 
  •  利用一个新的terminal打开正在运行的container

# watch the containers which are running

sudo dockers ps

# go into one conainer

sudo docker exec -it sharp_ardinghelli bash

技术分享

技术分享

以上是关于docker下使用caffe的命令记录的主要内容,如果未能解决你的问题,请参考以下文章

window10下caffe(gpu版)安装 问题记录

ubuntu下docker 安装 & 使用 记录

markdown [Docker] Docker片段列表和命令#linux #docker #snippets

Caffe入门学习规划

python Caffe片段

为 caffe 构建 GPU docker 映像时出错:不支持的 gpu 架构“compute_60”