Cuda support doesn’t work with this fix. I could send a patch that makes Cuda work, but I’m not sure that the dependencies are correct (or rather, I’m pretty sure that they aren’t). In that patch the tests suite doesn’t find Cuda, but it works after installation.

@dnaq Could you share how you got CUDA support to work? If I build pytorchWithCuda the resulting package seems to lack CUDA support.

@sveitser This is the nix-file that I use. It builds correctly, but for some reason the test-suite doesn't find cuda. It seems to work though.

I haven't sent a pull-request, since I don't really know why it works. It seems like if cuda/cudnn aren't in nativeBuildInputs then pytorchs build system builds everything without cuda support, but if cuda/cudnn aren't in propagatedBuildInputs, then pytorch can't load cuda/cudnn at runtime.

@dnaq Thank you! Can confirm that this builds with cuda support and that support is there after building,

>>> import torch
>>> torch.cuda.is_available()
True

Why not open a PR with this? It might be work in progress but is IMO much better than the current situation.

@dnaq Thanks for figuring that out. Can you create a PR please? Agree with @sveitser that it's better than the current situation which is broken for some users.