Pytorch cuda on WSL

Yes, I got it running. libcuda.so is in /usr/lib/wsl/lib and the problem is that nix doesn’t find it (pip must do special handling for wsl). No /run/opengl-driver/ directory exists for me.

I’m to sure how to add nixGL to my shell.nix file in the op… But I though that shouldn’t modify files outside /nix/store/. So may be nixGL needs to be add through nix-env which I though was only for nixOS…