Failed to build diff-gaussian-rasterization

[hxj2580]

root@autodl-container-eab742abf6-6d115628:~/autodl-fs/dreamgaussian-main# conda install pytorch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pytorch-cuda=11.8 -c pytorch -c nvidia Collecting package metadata (current_repodata.json): done Solving environment: failed with initial frozen solve. Retrying with flexible solve. Collecting package metadata (repodata.json): done Solving environment: | (wyn) root@autodl-container-eab742abf6-6d115628:~/autodl-fs/dreamgaussian-main# https://pytorch.org/get-started/previous-versions/ -bash: https://pytorch.org/get-started/previous-versions/: No such file or directory (wyn) root@autodl-container-eab742abf6-6d115628:~/autodl-fs/dreamgaussian-main# pip install ./diff-gaussian-rasterization Looking in indexes: https://pypi.tuna.tsinghua.edu.cn/simple Processing ./diff-gaussian-rasterization Preparing metadata (setup.py) ... done Building wheels for collected packages: diff-gaussian-rasterization Building wheel for diff-gaussian-rasterization (setup.py) ... error error: subprocess-exited-with-error

× python setup.py bdist_wheel did not run successfully. │ exit code: 1 ╰─> [54 lines of output] No CUDA runtime is found, using CUDA_HOME='/usr/local/cuda' running bdist_wheel running build running build_py creating build creating build/lib.linux-x86_64-cpython-38 creating build/lib.linux-x86_64-cpython-38/diff_gaussian_rasterization copying diff_gaussian_rasterization/init.py -> build/lib.linux-x86_64-cpython-38/diff_gaussian_rasterization running build_ext Traceback (most recent call last): File "", line 2, in File "", line 34, in File "/autodl-fs/data/dreamgaussian-main/diff-gaussian-rasterization/setup.py", line 17, in setup( File "/root/miniconda3/envs/wyn/lib/python3.8/site-packages/setuptools/init.py", line 103, in setup return distutils.core.setup(**attrs) File "/root/miniconda3/envs/wyn/lib/python3.8/site-packages/setuptools/_distutils/core.py", line 185, in setup return run_commands(dist) File "/root/miniconda3/envs/wyn/lib/python3.8/site-packages/setuptools/_distutils/core.py", line 201, in run_commands dist.run_commands() File "/root/miniconda3/envs/wyn/lib/python3.8/site-packages/setuptools/_distutils/dist.py", line 969, in run_commands self.run_command(cmd) File "/root/miniconda3/envs/wyn/lib/python3.8/site-packages/setuptools/dist.py", line 989, in run_command super().run_command(command) File "/root/miniconda3/envs/wyn/lib/python3.8/site-packages/setuptools/_distutils/dist.py", line 988, in run_command cmd_obj.run() File "/root/miniconda3/envs/wyn/lib/python3.8/site-packages/wheel/bdist_wheel.py", line 364, in run self.run_command("build") File "/root/miniconda3/envs/wyn/lib/python3.8/site-packages/setuptools/_distutils/cmd.py", line 318, in run_command self.distribution.run_command(command) File "/root/miniconda3/envs/wyn/lib/python3.8/site-packages/setuptools/dist.py", line 989, in run_command super().run_command(command) File "/root/miniconda3/envs/wyn/lib/python3.8/site-packages/setuptools/_distutils/dist.py", line 988, in run_command cmd_obj.run() File "/root/miniconda3/envs/wyn/lib/python3.8/site-packages/setuptools/_distutils/command/build.py", line 131, in run self.run_command(cmd_name) File "/root/miniconda3/envs/wyn/lib/python3.8/site-packages/setuptools/_distutils/cmd.py", line 318, in run_command self.distribution.run_command(command) File "/root/miniconda3/envs/wyn/lib/python3.8/site-packages/setuptools/dist.py", line 989, in run_command super().run_command(command) File "/root/miniconda3/envs/wyn/lib/python3.8/site-packages/setuptools/_distutils/dist.py", line 988, in run_command cmd_obj.run() File "/root/miniconda3/envs/wyn/lib/python3.8/site-packages/setuptools/command/build_ext.py", line 88, in run _build_ext.run(self) File "/root/miniconda3/envs/wyn/lib/python3.8/site-packages/setuptools/_distutils/command/build_ext.py", line 345, in run self.build_extensions() File "/root/miniconda3/envs/wyn/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 523, in build_extensions _check_cuda_version(compiler_name, compiler_version) File "/root/miniconda3/envs/wyn/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 414, in _check_cuda_version raise RuntimeError(CUDA_MISMATCH_MESSAGE.format(cuda_str_version, torch.version.cuda)) RuntimeError: The detected CUDA version (11.6) mismatches the version that was used to compile PyTorch (12.1). Please make sure to use the same CUDA versions.

  [end of output]

note: This error originates from a subprocess, and is likely not a problem with pip. ERROR: Failed building wheel for diff-gaussian-rasterization Running setup.py clean for diff-gaussian-rasterization Failed to build diff-gaussian-rasterization ERROR: Could not build wheels for diff-gaussian-rasterization, which is required to install pyproject.toml-based projects

What can i do？

[inuex35]

Maybe the problem is here. The detected CUDA version (11.6) mismatches the version that was used to compile PyTorch (12.1). Please make sure to use the same CUDA versions. Check your torch and cuda version compatibility.

[mayankysharma]

Yes, even I faced this issue and to solve this I installed CUDA version(11.6) which is mentioned in the requirements.

[tslgithub]

  The detected CUDA version (11.6) mismatches the version that was used to compile
  PyTorch (12.1). Please make sure to use the same CUDA versions.