search

Colmar-zlicheng / Spring-Gaus

[ECCV 2024] Reconstruction and Simulation of Elastic Objects with Spring-Mass 3D Gaussians
https://zlicheng.com/spring_gaus/
Apache License 2.0
63 stars 0 forks source link

ModuleNotFoundError: No module named 'torch._custom_ops' #3

Open dangzhenmo opened 5 months ago

dangzhenmo commented 5 months ago

The ERROR occur when I try to train the data.I was using CUDA=11.8 and pytorch3D==0.7.4,I don't sure whether this is a potential reason or not. The full ERROR report are as follow: (base) PS F:\hhy\Spring-Gaus-main> python train.py -g 0 --cfg config/real_capture/burger.yaml --eval_cam 0 --exp_id burger Traceback (most recent call last): File "F:\hhy\Spring-Gaus-main\train.py", line 30, in from lib.utils.transform import inverse_sigmoid, uniform_sampling File "F:\hhy\Spring-Gaus-main\lib\utils\transform.py", line 11, in import torchvision.transforms.functional as tvF File "C:\Users\Administrator\anaconda3\Lib\site-packages\torchvision__init__.py", line 6, in from torchvision import _meta_registrations, datasets, io, models, ops, transforms, utils File "C:\Users\Administrator\anaconda3\Lib\site-packages\torchvision_meta_registrations.py", line 4, in import torch._custom_ops ModuleNotFoundError: No module named 'torch._custom_ops'

dangzhenmo commented 5 months ago

also I'm using python=3.11 in the base env

Colmar-zlicheng commented 5 months ago

Hi, it looks like a package version problem, maybe try reinstall pytorch and torchvision can help.

You can use the recommend environment at https://github.com/Colmar-zlicheng/Spring-Gaus?tab=readme-ov-file#rocket-dependencies

sLeency commented 4 months ago

I have the same problem.Have you solved it?

sLeency commented 4 months ago

I tried reinstalling torch and torchvision,but it didn't work.

Colmar-zlicheng commented 4 months ago

I tried reinstalling torch and torchvision,but it didn't work.

Can you provide your torch version?

dangzhenmo commented 4 months ago

I didn't solve the problem.

Colmar-zlicheng commented 4 months ago

I didn't solve the problem.

What's your pytorch version and torchvision version?

dangzhenmo commented 4 months ago

Hi, it looks like a package version problem, maybe try reinstall pytorch and torchvision can help.

You can use the recommend environment at https://github.com/Colmar-zlicheng/Spring-Gaus?tab=readme-ov-file#rocket-dependencies Thank you vert much.I guess maybe the main reason of error is I didn't install pytorch3D==0.7.1 which is supported for python==3.7 and CUDA==11.6 at the same time. But I only found the linux download source from here:https://anaconda.org/pytorch3d/pytorch3d/files?page=7 Since I am using windows and my homework was ended I give up on this project.

dangzhenmo commented 4 months ago

It seems that windows only supported pytorch3D==0.7.4 and it is no longer support python==3.7