{ pkgs ? import {config = { allowUnfree = false; cudaSupport = false;}; }}: pkgs.mkShell { buildInputs = with pkgs; [ (python311.withPackages(ps: with ps; [ pip numpy notebook tkinter pygame-ce ])) ]; #cudaPackages.cudatoolkit shellHook = '' echo "You are now using a NIX environment" idle3.11 ''; }