From c8d4c49fb8e89f6954fbacce077a60d507a798eb Mon Sep 17 00:00:00 2001
From: David Crompton <david.crompton@mail.utoronto.ca>
Date: Tue, 5 Nov 2024 12:08:02 -0500
Subject: [PATCH] KCNHUB: Python/Nvidia: Add packages for pycuda compat
 (gcc11/cuda inc. path)

---
 machines/kcnhub/nvidia.nix | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/machines/kcnhub/nvidia.nix b/machines/kcnhub/nvidia.nix
index 822c1d4..64ba0d8 100644
--- a/machines/kcnhub/nvidia.nix
+++ b/machines/kcnhub/nvidia.nix
@@ -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 = [