ENet

Install

Obtain ENet

git clone --recursive https://github.com/TimoSaemann/ENet.git
cd ENet/caffe-enet/docker

Config caffe-enet docker

vim Makefile:

cpu_standalone: standalone/cpu/Dockerfile
> $(DOCKER) build -t caffe-enet:cpu standalone/cpu

gpu_standalone: standalone/gpu/Dockerfile
 $(DOCKER) build -t caffe-enet:gpu standalone/gpu

FROM_GPU = "nvidia/cuda:8.0-cudnn5-devel-ubuntu16.04"
FROM_CPU = "ubuntu:16.04"

vim standalone/gpu/Dockerfile:

FROM nvidia/cuda:8.0-cudnn5-devel-ubuntu16.04

I think it is the same as

FROM nvidia/cuda:8.0-cudnn5-devel

vim standalone/cpu/Dockerfile:

FROM ubuntu:16.04

Build caffe-enet

make all
or
docker build -t caffe-enet:gpu standalone/gpu #or cpu

Check docker image:

~/Projects/caffe-enet-docker$ sudo docker image ls
REPOSITORY          TAG                            IMAGE ID            CREATED             SIZE
caffe-enet          gpu                            7e2d7b05a1ca        19 minutes ago      3.05GB
nvidia/cuda         8.0-cudnn5-devel-ubuntu16.04   38622aa43f13        5 days ago          1.85GB

Refer https://github.com/TimoSaemann/ENet/tree/master/Tutorial

Check version

ENet/caffe-enet/docker$ sudo docker run -ti caffe-enet:gpu caffe --v
ersion
caffe version 1.0.0

Run test:

 sudo nvidia-docker run -ti caffe-enet:gpu bash -c "cd /opt/caffe/build; make runtest"

Reference:

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值