How to properly setup an environment with Pytorch?

CUDA packages are not cached in the main cache.nixos.org due to the licensing situation. I think there’s a community cache available right now, and a new initiative happening to improve things but I’m not too sure on the specifics. There was a recent announcement. Nix, Flox, Nvidia Opening Up CUDA Redistribution on Nix

1 Like