[BUG] Running demo.py causes a CUDA error

[acm94994]

Operating system: Ubuntu 22.04 LTS
Python 3.7 (the environment provided was installed, Torch 1.4)

Running the demo caused a CUDA runtime error.

"no kernel image is available for execution on the device at /pytorch/aten/src/THCUNN/generic/LeakyReLU.cu:29"

print(torch.version) -> 1.4.0

print(torch.version.cuda) -> 10.1

[sky1614]

@acm94994 I would like to work on this issue. Can you please assign it to me?

[needabetterusername]

@sky1614 Any progress on this?

I am trying to run this on a 4090 with CUDA 12.9. Sticking with python 3.7 enforces a Torch version which does not support Ada Lovelace arch.

I didn't get immediate success with python 3.8... still investigating.