45 lines
1.3 KiB
Nix
45 lines
1.3 KiB
Nix
{
|
|
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;
|
|
};
|
|
}
|