Skip to content

MemoryError: cudaErrorMemoryAllocation #246

@ItsReptile

Description

@ItsReptile

I am on WSL, I had quite a few problems trying to install everything considering that I am using a 5060Ti (sm_120), I had to install CUDA 12.8, I hope that's not the problem.
When I run this, I consistently get this error, at start I was using a model of 1000+ images, so I considered it reasonable (for now), but after trying with a 34 images one, i still get that error, any fixes?

python train_full_pipeline.py -s "/mnt/c/users/esimple/desktop/data/cubo/" -r "dn_consistency" --high_poly False --export_obj True

output:

Will export a UV-textured mesh as an .obj file.
Will export a ply file with the refined 3D Gaussians at the end of the training.
Optimizing output/vanilla_gs/cubo/
Output folder: output/vanilla_gs/cubo/ [16/07 12:12:51]
Reading camera 34/34 [16/07 12:12:51]
Loading Training Cameras [16/07 12:12:51]
Loading Test Cameras [16/07 12:12:51]
Number of points at initialisation :  2718 [16/07 12:12:51]
Traceback (most recent call last):
  File "/mnt/c/Users/esimple/Desktop/SuGaR/./gaussian_splatting/train.py", line 220, in <module>
    training(lp.extract(args), op.extract(args), pp.extract(args), args.test_iterations, args.save_iterations, args.checkpoint_iterations, args.start_checkpoint, args.debug_from)
  File "/mnt/c/Users/esimple/Desktop/SuGaR/./gaussian_splatting/train.py", line 36, in training
    scene = Scene(dataset, gaussians)
  File "/mnt/c/Users/esimple/Desktop/SuGaR/gaussian_splatting/scene/__init__.py", line 83, in __init__
    self.gaussians.create_from_pcd(scene_info.point_cloud, self.cameras_extent)
  File "/mnt/c/Users/esimple/Desktop/SuGaR/gaussian_splatting/scene/gaussian_model.py", line 134, in create_from_pcd
    dist2 = torch.clamp_min(distCUDA2(torch.from_numpy(np.asarray(pcd.points)).float().cuda()), 0.0000001)
MemoryError: std::bad_alloc: cudaErrorMemoryAllocation: out of memory

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions