KCNHUB: Python/Nvidia: Add packages for pycuda compat (gcc11/cuda inc. path)

master
David Crompton 2024-11-05 12:08:02 -05:00
parent 6f94305606
commit c8d4c49fb8
1 changed files with 9 additions and 0 deletions

View File

@ -37,8 +37,17 @@
# Add cuda to system environment
environment.systemPackages = with pkgs; [
cudatoolkit
# Cudatoolkit by default only supports gcc11 for our version
gcc11
];
# Some packages (such as pyCuda expect this include path to be exposed)
environment.sessionVariables = {
CPATH = [
"${pkgs.cudatoolkit}/include"
];
};
# Use Prebuilt Cached Versions of CUDA Packages
nix.settings = {
substituters = [