svox2 icon indicating copy to clipboard operation
svox2 copied to clipboard

AssertionError: CUDA extension is currently required for accelerate

Open puallee opened this issue 2 years ago • 8 comments

I still have the probelm of AssertionError: CUDA extension is currently required for accelerate ubuntu@ubuntu-broncus:/erick/rendering/svox2-master/svox2-master/opt$ python3.8 render_imgs.py /home/raid/erickData/data/TanksAndTempleBG/Train/ ../svox2/utils.py:39: UserWarning: CUDA extension svox2.csrc could not be loaded! Operations will be slow. Please do not import svox in the svox2 source directory. warn("CUDA extension svox2.csrc could not be loaded! " + Setting up [LPIPS] perceptual loss: trunk [vgg], v[0.1], spatial [off] Loading model from: /home/ubuntu/.local/lib/python3.8/site-packages/lpips/weights/v0.1/vgg.pth Defaulting to extended NSVF dataset LOAD NSVF DATA /home/raid/erickData/data/TanksAndTempleBG/Train/ split test 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 43/43 [00:00<00:00, 58.04it/s] NORMALIZE BY? camera scene_scale 2.0069090349061196 intrinsics (loaded reso) Intrin(fx=580.62451171875, fy=580.62451171875, cx=490.75, cy=273.0) Traceback (most recent call last): File "render_imgs.py", line 116, in grid = svox2.SparseGrid.load(args.ckpt, device=device) File "../svox2/svox2.py", line 1629, in load grid.accelerate() File "../svox2/svox2.py", line 1491, in accelerate assert ( AssertionError: CUDA extension is currently required for accelerate Can some help me ,my environment is python3.8 torch 1.8.0+cu111 nvidia driver 460.91.03 gpu 3090 ,torchaudio 0.8.0 torchvision 0.9.0+cu111 and gcc version 7.5.0 (Ubuntu 7.5.0-3ubuntu118.04)

puallee avatar Jan 29 '22 02:01 puallee