RuntimeError: CUDA error: invalid device function

[GeorvityLabs]

Hey guys,

I made a colab notebook for this project.
I'm currently getting this error

Traceback (most recent call last): File "demo.py", line 341, in <module> model = Demo(args) File "demo.py", line 53, in __init__ self.renderer_mr = Renderer(cfgs, im_size=128) File "/content/3DFaceReconstruction-LAP/lap/renderer/renderer_mr.py", line 45, in __init__ self.inv_k_mat = torch.inverse(k_mat).unsqueeze(0) RuntimeError: CUDA error: invalid device function

How to fix this?

Here is a link to my notebook , you can make a copy and edit it

https://colab.research.google.com/drive/18tIkvLIaN-_v3sLW2ykkcLGpqxwRN13z?usp=sharing

[cuizhanyuan]

Just delete." import torch; torch.version".

[GeorvityLabs]

@cuizhanyuan that doesn't fix the error.

Still getting :
Traceback (most recent call last):
File "demo.py", line 341, in
model = Demo(args)
File "demo.py", line 53, in init
self.renderer_mr = Renderer(cfgs, im_size=128)
File "/content/3DFaceReconstruction-LAP/lap/renderer/renderer_mr.py", line 45, in init
self.inv_k_mat = torch.inverse(k_mat).unsqueeze(0)
RuntimeError: CUDA error: invalid device function

Link to colab , please run and check it yourself!
https://colab.research.google.com/drive/18tIkvLIaN-_v3sLW2ykkcLGpqxwRN13z?usp=sharing

[cuizhanyuan]

I don't know what to do. I commented out that part yesterday and it worked. Today it failed.