tdpeuter/flakes
tdpeuter/flakes
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

feat(python): Add CUDA

Browse source
This commit is contained in:
Tibo De Peuter 2025-12-06 20:21:43 +01:00
parent 6d91d83d86
commit 595d152885
Signed by: tdpeuter
GPG key ID: 38297DE43F75FFE2
1 changed files with 16 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

19
python/uv/flake.nix
View file

@ -31,6 +31,15 @@
devShells.default = pkgs.mkShell { devShells.default = pkgs.mkShell {
packages = (with pkgs; [ packages = (with pkgs; [
python311Full python311Full
# CUDA
git gitRepo gnupg autoconf curl
procps gnumake util-linux m4 gperf unzip
cudatoolkit linuxPackages.nvidia_x11
libGLU libGL
xorg.libXi xorg.libXmu freeglut
xorg.libXext xorg.libX11 xorg.libXv xorg.libXrandr zlib
ncurses5 stdenv.cc binutils
]) ++ (with pkgs-unstable; [ ]) ++ (with pkgs-unstable; [
uv uv
]) ++ (with nix-jetbrains-plugins.lib."${system}"; [ ]) ++ (with nix-jetbrains-plugins.lib."${system}"; [
@ -38,11 +47,15 @@
(buildIdeWithPlugins pkgs-unstable.jetbrains "pycharm-professional" pluginList) (buildIdeWithPlugins pkgs-unstable.jetbrains "pycharm-professional" pluginList)
]); ]);
# CUDA
CUDA_PATH = pkgs.cudatoolkit;
# ImportError: libstdc++.so.6: cannot open shared object file: No such file or directory
LD_LIBRARY_PATH = "${pkgs.linuxPackages.nvidia_x11}/lib:${pkgs.ncurses5}/lib:${pkgs.libGL}/lib/:${pkgs.stdenv.cc.cc.lib}/lib/:${pkgs.glibc}/lib";
EXTRA_LDFLAGS = "-L/lib -L${pkgs.linuxPackages.nvidia_x11}/lib";
EXTRA_CCFLAGS = "-I/usr/include";
# Stop uv from downloading Python binaries automatically if needed. # Stop uv from downloading Python binaries automatically if needed.
UV_PYTHON_DOWNLOADS = "never"; UV_PYTHON_DOWNLOADS = "never";
# ImportError: libstdc++.so.6: cannot open shared object file: No such file or directory
LD_LIBRARY_PATH = "${pkgs.stdenv.cc.cc.lib}/lib";
}; };
}); });
} }