fyu / dilation

Dilated Convolution for Semantic Image Segmentation

Home Page:https://www.vis.xyz/pub/dilation

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How much GPU Memory is required/recommended to run the demo?

kaisark opened this issue · comments

How much GPU memory is required/recommended to run the demo? I am trying to run the demo on my Nvidia Jetson TX1 with 4 GB of RAM and the program terminates ("Killed") and/or reboots the machine.

cuda 8
cudnn 6
ubuntu 16

#1:
python predict.py pascal_voc images/dog.jpg --gpu 0

#2:
python predict.py kitti images/example_kitti.png --gpu 0

nvidia@tegra-ubuntu:~/cviz/dilation$ python predict.py kitti images/example_kitti.png --gpu 0
I0213 00:39:05.451731 2439 gpu_memory.cpp:159] GPUMemory::Manager initialized with Caching (CUB) GPU Allocator
I0213 00:39:05.451974 2439 gpu_memory.cpp:161] Total memory: 4174815232, Free: 1888354304, dev_info[0]: total=4174815232 free=1888354304
I0213 00:39:05.452177 2439 gpu_memory.cpp:159] GPUMemory::Manager initialized with Caching (CUB) GPU Allocator
I0213 00:39:05.452195 2439 gpu_memory.cpp:161] Total memory: 4174815232, Free: 1888354304, dev_info[0]: total=4174815232 free=1888354304
Using GPU 0
I0213 00:39:05.463042 2439 upgrade_proto.cpp:66] Attempting to upgrade input file specified using deprecated input fields: models/dilation7_kitti_deploy.prototxt
I0213 00:39:05.463099 2439 upgrade_proto.cpp:69] Successfully upgraded file specified using deprecated input fields.
W0213 00:39:05.463116 2439 upgrade_proto.cpp:71] Note that future Caffe releases will only support input layers and not input fields.
I0213 00:39:05.463681 2439 net.cpp:70] Initializing net from parameters:
state {