-
Notifications
You must be signed in to change notification settings - Fork 260
Open
Description
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
Labels
No labels