KCNHUB: Python/Nvidia: Add packages for pycuda compat (gcc11/cuda inc. path)
							parent
							
								
									6f94305606
								
							
						
					
					
						commit
						c8d4c49fb8
					
				| 
						 | 
				
			
			@ -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 = [
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
		Reference in New Issue