Pytorch cuda on WSL

[tom-huntington]

I found the solution here CUDA support running on WSL · Issue #535 · hasktorch/hasktorch · GitHub

you can add

shellHook = ''
    export LD_LIBRARY_PATH=/usr/lib/wsl/lib
  '';