search

wanmeihuali / taichi_3d_gaussian_splatting

An unofficial implementation of paper 3D Gaussian Splatting for Real-Time Radiance Field Rendering by taichi lang.
Apache License 2.0
637 stars 58 forks source link

Torch not compiled with CUDA enabled #118

Closed kenculate closed 10 months ago

kenculate commented 10 months ago

I am trying to run just the visualizer, and facing these error

python visualizer.py --parquet_path_list boots_19000.parquet

[Taichi] version 1.6.0, llvm 15.0.1, commit f1c6fbbd, win, python 3.10.11 C:\Users\saeid\source\repos\taichi_3d_gaussian_splatting\taichi_3d_gaussian_splatting\GaussianPointCloudScene.py:6: DeprecationWarning: Please use cKDTree from the scipy.spatial namespace, the scipy.spatial.ckdtree namespace is deprecated. from scipy.spatial.ckdtree import cKDTree Traceback (most recent call last): File "C:\Users\saeid\source\repos\taichi_3d_gaussian_splatting\visualizer.py", line 23, in class GaussianPointVisualizer: File "C:\Users\saeid\source\repos\taichi_3d_gaussian_splatting\visualizer.py", line 25, in GaussianPointVisualizer class GaussianPointVisualizerConfig: File "C:\Users\saeid\source\repos\taichi_3d_gaussian_splatting\visualizer.py", line 29, in GaussianPointVisualizerConfig camera_intrinsics: torch.Tensor = torch.tensor( File "C:\Users\saeid\source\repos\taichi_3d_gaussian_splatting\venv\lib\site-packages\torch\cuda__init__.py", line 239, in _lazy_init raise AssertionError("Torch not compiled with CUDA enabled") AssertionError: Torch not compiled with CUDA enabled

wanmeihuali commented 10 months ago

Please install torch with CUDA... e.g. 

conda install pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.7 -c pytorch -c nvidia

please also ensure your cuda version is correct.