{ description = "Python 3.10 development environment"; outputs = { self, nixpkgs }: let system = "x86_64-linux"; pkgs = import nixpkgs { inherit system; config.allowUnfree = true; }; in { devShells.${system}.default = (pkgs.buildFHSEnv { name = "nvidia-fuck-you"; targetPkgs = pkgs: (with pkgs; [ python310 cudaPackages_12.cudatoolkit python310Packages.pytorch-bin python310Packages.pip python310Packages.sphinx gcc linuxPackages.nvidia_x11 libGLU libGL xorg.libXi xorg.libXmu freeglut xorg.libXext xorg.libX11 xorg.libXv xorg.libXrandr zlib ncurses5 stdenv.cc binutils ffmpeg # I daily drive the fish shell # you can remove this, the default is bash fish ]); profile = '' export LD_LIBRARY_PATH="${pkgs.linuxPackages.nvidia_x11}/lib" export CUDA_PATH="${pkgs.cudatoolkit}" export EXTRA_LDFLAGS="-L/lib -L${pkgs.linuxPackages.nvidia_x11}/lib" export EXTRA_CCFLAGS="-I/usr/include" ''; # again, you can remove this if you like bash runScript = "fish"; }).env; }; }