No CUDA runtime is found & ModuleNotFoundError: No module named '_gridencoder'

[jygkh]

(venv_stable-dreamfusion) PS D:\python\stable-dreamfusion> python main.py --text "a hamburger" --workspace trial -O No CUDA runtime is found, using CUDA_HOME='C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1' Namespace(file=None, text='a hamburger', negative='', O=True, O2=False, test=False, six_views=False, eval_interval=1, test_interval=100, workspace='trial', seed=None, image=None, image_config=None, known_view_interval=4, IF=False, guidance=['SD'], guidance_scale=100, save_mesh=False, mcubes_resolution=256, decimate_target=50000.0, dmtet=False, tet_grid_size=128, init_with='', lock_geo=False, perpneg=False, negative_w=-2, front_decay_factor=2, side_decay_factor=10, iters=10000, lr=0.001, ckpt='latest', cuda_ray=True, taichi_ray=False, max_steps=1024, num_steps=64, upsample_steps=32, update_extra_interval=16, max_ray_batch=4096, latent_iter_ratio=0.2, albedo_iter_ratio=0, min_ambient_ratio=0.1, textureless_ratio=0.2, jitter_pose=False, jitter_center=0.2, jitter_target=0.2, jitter_up=0.02, uniform_sphere_rate=0, grad_clip=-1, grad_clip_rgb=-1, bg_radius=1.4, density_activation='exp', density_thresh=10, blob_density=5, blob_radius=0.2, backbone='grid', optim='adan', sd_version='2.1', hf_key=None, fp16=True, vram_O=False, w=64, h=64, known_view_scale=1.5, known_view_noise_scale=0.002, dmtet_reso_scale=8, batch_size=1, bound=1, dt_gamma=0, min_near=0.01, radius_range=[3.0, 3.5], theta_range=[45, 105], phi_range=[-180, 180], fovy_range=[10, 30], default_radius=3.2, default_polar=90, default_azimuth=0, default_fovy=20, progressive_view=False, progressive_view_init_ratio=0.2, progressive_level=False, angle_overhead=30, angle_front=60, t_range=[0.02, 0.98], dont_override_stuff=False, lambda_entropy=0.001, lambda_opacity=0, lambda_orient=0.01, lambda_tv=0, lambda_wd=0, lambda_mesh_normal=0.5, lambda_mesh_laplacian=0.5, lambda_guidance=1, lambda_rgb=1000, lambda_mask=500, lambda_normal=0, lambda_depth=10, lambda_2d_normal_smooth=0, lambda_3d_normal_smooth=0, save_guidance=False, save_guidance_interval=10, gui=False, W=800, H=800, radius=5, fovy=20, light_theta=60, light_phi=0, max_spp=1, zero123_config='./pretrained/zero123/sd-objaverse-finetune-c_concat-256.yaml', zero123_ckpt='pretrained/zero123/zero123-xl.ckpt', zero123_grad_scale='angle', dataset_size_train=100, dataset_size_valid=8, dataset_size_test=100, exp_start_iter=0, exp_end_iter=10000, images=None, ref_radii=[], ref_polars=[], ref_azimuths=[], zero123_ws=[], default_zero123_w=1) Traceback (most recent call last): File "D:\python\stable-dreamfusion\gridencoder\grid.py", line 11, in import _gridencoder as _backend ModuleNotFoundError: No module named '_gridencoder'

During handling of the above exception, another exception occurred:

Traceback (most recent call last): File "D:\python\stable-dreamfusion\main.py", line 315, in model = NeRFNetwork(opt).to(device) File "D:\python\stable-dreamfusion\nerf\network_grid.py", line 49, in init self.encoder, self.in_dim = get_encoder('hashgrid', input_dim=3, log2_hashmap_size=19, desired_resolution=2048 * self.bound, interpolation='smoothstep') File "D:\python\stable-dreamfusion\encoding.py", line 75, in get_encoder from gridencoder import GridEncoder File "D:\python\stable-dreamfusion\gridencoder__init__.py", line 1, in from .grid import GridEncoder File "D:\python\stable-dreamfusion\gridencoder\grid.py", line 13, in from .backend import _backend File "D:\python\stable-dreamfusion\gridencoder\backend.py", line 31, in _backend = load(name='_grid_encoder', File "D:\python\stable-dreamfusion\venv_stable-dreamfusion\lib\site-packages\torch\utils\cpp_extension.py", line 1308, in load return _jit_compile( File "D:\python\stable-dreamfusion\venv_stable-dreamfusion\lib\site-packages\torch\utils\cpp_extension.py", line 1710, in _jit_compile _write_ninja_file_and_build_library( File "D:\python\stable-dreamfusion\venv_stable-dreamfusion\lib\site-packages\torch\utils\cpp_extension.py", line 1810, in _write_ninja_file_and_build_library _write_ninja_file_to_build_library( File "D:\python\stable-dreamfusion\venv_stable-dreamfusion\lib\site-packages\torch\utils\cpp_extension.py", line 2199, in _write_ninja_file_to_build_library cuda_flags = common_cflags + COMMON_NVCC_FLAGS + _get_cuda_arch_flags() File "D:\python\stable-dreamfusion\venv_stable-dreamfusion\lib\site-packages\torch\utils\cpp_extension.py", line 1980, in _get_cuda_arch_flags arch_list[-1] += '+PTX' IndexError: list index out of range (venv_stable-dreamfusion) PS D:\python\stable-dreamfusion> nvcc --version nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2023 NVIDIA Corporation Built on Wed_Feb__8_05:53:42_Coordinated_Universal_Time_2023 Cuda compilation tools, release 12.1, V12.1.66 Build cuda_12.1.r12.1/compiler.32415258_0

[DSaurus]

Hi @jygkh ,

This error is not related to threestudio and will be closed.