{ description = "A very basic flake"; inputs = { nixpkgs.url = "github:nixos/nixpkgs/nixos-unstable"; flake-utils.url = "github:numtide/flake-utils"; }; outputs = { self, nixpkgs, flake-utils }: flake-utils.lib.eachDefaultSystem (system: let pkgs = import nixpkgs { inherit system; config = { allowUnfree = true; }; }; in rec { overlays = [(final: prev: { pythonPackagesExtensions = [(py-final: py-prev: { torch = py-final.torch-bin; torchvision = py-final.torchvision-bin; torchaudio = py-final.torchaudio-bin; })]; })]; devShell = pkgs.mkShell { buildInputs = with pkgs; [ (python3.withPackages(ps: with ps; [ ipython jupyter numpy pandas matplotlib torch scikit-learn timeout-decorator torchvision seaborn ])) ]; }; } ); }