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