search

gsgen3d / gsgen

[CVPR 2024] Text-to-3D using Gaussian Splatting
https://arxiv.org/abs/2309.16585
MIT License
761 stars 42 forks source link

RuntimeError: p1 must be a CUDA tensor. #23

Open Coder-ZZY opened 10 months ago

Coder-ZZY commented 10 months ago

When I try to export mesh using this script: python utils/export.py checkpoints/a_high_quality_photo_of_a_corgi/2023-10-24/140914/ckpts/step_14000.pt --type mesh --batch_size 65536 --reso 256 --K 200 --thresh 0.1 I meet this error:

  File "/mnt/sda//gsgen/utils/export.py", line 138, in to_mesh
    density_val_grid, L = get_density_val_grid_from_ckpt(
  File "/mnt/sda//gsgen/utils/export.py", line 94, in get_density_val_grid_from_ckpt
    _, nn_idx, dist = K_nearest_neighbors(
  File "/mnt/sda//anaconda3/envs/gsgen/lib/python3.9/site-packages/torch/autograd/grad_mode.py", line 27, in decorate_context
    return func(*args, **kwargs)
  File "/mnt/sda//gsgen/utils/ops.py", line 129, in K_nearest_neighbors
    dist, idx, nn = knn_points(query[None, ...], mean[None, ...], K=K, return_nn=True)
  File "/mnt/sda//anaconda3/envs/gsgen/lib/python3.9/site-packages/pytorch3d/ops/knn.py", line 187, in knn_points
    p1_dists, p1_idx = _knn_points.apply(
  File "/mnt/sda//anaconda3/envs/gsgen/lib/python3.9/site-packages/pytorch3d/ops/knn.py", line 72, in forward
    idx, dists = _C.knn_points_idx(p1, p2, lengths1, lengths2, norm, K, version)
RuntimeError: p1 must be a CUDA tensor.
Moondok commented 2 weeks ago

@Coder-ZZY hi, have you solved this problem, I met it also.