search

zju3dv / animatable_nerf

Code for "Animatable Implicit Neural Representations for Creating Realistic Avatars from Videos" TPAMI 2024, ICCV 2021
Other
494 stars 50 forks source link

RuntimeError: culabs runtime error: the GPU program failed to execute at /pytorch/aten/src/THC/THCBlas.cu:368 #21

Closed 3DebbieChen9 closed 2 years ago

3DebbieChen9 commented 2 years ago

While I was trying to visualize novel views of the 0-th frame on Human3.6M. I change a way to install the PyTorch, but I still encounter the following error... RuntimeError: cublas runtime error : the GPU program failed to execute at /pytorch/aten/src/THC/THCBlas.cu:368 image

I use pip install torch==1.4.0 torchvision==0.5.0 to install the torch (Exactly like the way that was described on the PyTorch Website)

image

And my set-up is

I think that the PyTorch version do match with the system cuda. Does anyone have encountered the same problem and solved it?

pengsida commented 2 years ago

Maybe it is caused by this problem: https://github.com/pytorch/pytorch/issues/12977