How to set up minimal environment with Pytorch and avoid long compile times

Hi all, I’m relatively new to Nix and NixOS and am trying to create a minimal shell.nix to create a shell environment in which I can run Python and Pytorch. I followed Declarative shell environments with shell.nix — nix.dev documentation to create the following shell.nix: let nixpkgs = fetchTarball "https://github.com/NixOS/nixpkgs/archive/8b8c811c7c2541c30382c5de7ed26be055569c60.tar.gz"; pkgs = import nixpkgs { config = { allowUnfree = true; }; overlays = []; }; in pkgs.mkShellNoCC ...