Votenet on Nvidia Jetson Nano 4GB

facebookresearch / votenet

Deep Hough Voting for 3D Object Detection in Point Clouds

MIT License

1.69k stars 377 forks source link

Votenet on Nvidia Jetson Nano 4GB #167

Open lschoole opened 1 year ago

lschoole commented 1 year ago

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 commented 1 year ago

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)