Votenet on Nvidia Jetson Nano 4GB

[lschoole]

Hi,
I have gotten to training the VoteNet model on SUN RGB D data, but python3 train.py (--use_sunrgbd_v2) returns "Killed" without ever getting through the first forward pass.

[lschoole]

I should mention that I do have Cuda installed, but I am seeing that the MB memory is very small.

I tensorflow/core/common_runtime/gpu/gpu_device.cc:1406] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 255 MB memory) -> physical GPU (device: 0, name: NVIDIA Tegra X1, pci bus id: 0000:00:00.0, compute capability: 5.3)