Hi @ptrblck
I extracted a snippet of code. It’s a GAN for image translation (the current example is meaningless since it uses fake inputs).
In this case, if I run it on the CPU it works, if I switch to the GPU it crashes.
The toy code is this (I tried to paste it here but the indentation is not good):
https://pastebin.pl/view/551b716c
I’m using the weels for 1.8.1, but my nvcc version is 11.2 (not 11.1).
Thanks a lot for helping, I really appreciate it.