KCNHUB
/
WS-KCNHUB
1
0
Fork 0
Code Issues 2 Pull Requests Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
machines/kcnhub/nvidia.nix
View File

@ -37,8 +37,17 @@
# Add cuda to system environment # Add cuda to system environment
environment.systemPackages = with pkgs; [ environment.systemPackages = with pkgs; [
cudatoolkit 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 # Use Prebuilt Cached Versions of CUDA Packages
nix.settings = { nix.settings = {
substituters = [ substituters = [