Gsplat 설치할 때 No module named "torch" 뜨는 문제 (torch 설치 이미 되어있음)

error: subprocess-exited-with-error

  × Getting requirements to build wheel did not run successfully.
  │ exit code: 1
  ╰─> [22 lines of output]
      Setting MAX_JOBS to 10
      Traceback (most recent call last):
        File "/home/jseob/miniconda3/envs/da3/lib/python3.12/site-packages/pip/_vendor/pyproject_hooks/_in_process/_in_process.py", line 389, in <module>
          main()
        File "/home/jseob/miniconda3/envs/da3/lib/python3.12/site-packages/pip/_vendor/pyproject_hooks/_in_process/_in_process.py", line 373, in main
          json_out["return_val"] = hook(**hook_input["kwargs"])
                                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        File "/home/jseob/miniconda3/envs/da3/lib/python3.12/site-packages/pip/_vendor/pyproject_hooks/_in_process/_in_process.py", line 143, in get_requires_for_build_wheel
          return hook(config_settings)
                 ^^^^^^^^^^^^^^^^^^^^^
        File "/tmp/pip-build-env-sbqrwjfo/overlay/lib/python3.12/site-packages/setuptools/build_meta.py", line 331, in get_requires_for_build_wheel
          return self._get_build_requires(config_settings, requirements=[])
                 ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        File "/tmp/pip-build-env-sbqrwjfo/overlay/lib/python3.12/site-packages/setuptools/build_meta.py", line 301, in _get_build_requires
          self.run_setup()
        File "/tmp/pip-build-env-sbqrwjfo/overlay/lib/python3.12/site-packages/setuptools/build_meta.py", line 512, in run_setup
          super().run_setup(setup_script=setup_script)
        File "/tmp/pip-build-env-sbqrwjfo/overlay/lib/python3.12/site-packages/setuptools/build_meta.py", line 317, in run_setup
          exec(code, locals())
        File "<string>", line 135, in <module>
        File "<string>", line 33, in get_extensions
      ModuleNotFoundError: No module named 'torch'
      [end of output]

  note: This error originates from a subprocess, and is likely not a problem with pip.
  ERROR: Failed to build 'gsplat' when getting requirements to build wheel

 

gsplat 설치할 때 위와 같이 뜬금없이 torch 에러를 겪을 일이 있다. 가상 환경에 분명 torch 설치는 잘 되어있고 import도 문제없이 잘 되는 상태인데 반복돼서 까다로웠던 문제.

 

해결법

pip install ninja

일단 ninja 깔려있는건 확인해야 함. CUDA extension 빌드할 때 ninja가 필요하기 때문에 이게 없으면 에러가 날 수도 있음.

 

pip install git+https://github.com/nerfstudio-project/gsplat.git@0b4dddf04cb687367602c01196913cde6a743d70 --no-build-isolation

--no-build-islation 태그를 추가해서 설치해줘야 함. 이게 없으면 torch가 어디에 설치되어있는지 못 찾아서 위와 같은 문제가 날 수 있음.