Skip to content

Commit e2f071f

Browse files
authored
gpt4all-cuda: fix CUDA initialization (#342251)
2 parents 45b9542 + 06c5b53 commit e2f071f

File tree

1 file changed

+9
-6
lines changed

1 file changed

+9
-6
lines changed

pkgs/by-name/gp/gpt4all/package.nix

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
, wayland
1212
, cudaSupport ? config.cudaSupport
1313
, cudaPackages ? { }
14+
, autoAddDriverRunpath
1415
}:
1516

1617
stdenv.mkDerivation (finalAttrs: {
@@ -41,6 +42,7 @@ stdenv.mkDerivation (finalAttrs: {
4142
qt6.wrapQtAppsHook
4243
] ++ lib.optionals cudaSupport [
4344
cudaPackages.cuda_nvcc
45+
autoAddDriverRunpath
4446
];
4547

4648
buildInputs = [
@@ -56,12 +58,13 @@ stdenv.mkDerivation (finalAttrs: {
5658
vulkan-headers
5759
wayland
5860
] ++ lib.optionals cudaSupport (
59-
with cudaPackages;
60-
[
61-
cuda_cccl
62-
cuda_cudart
63-
libcublas
64-
]);
61+
with cudaPackages;
62+
[
63+
cuda_cccl
64+
cuda_cudart
65+
libcublas
66+
]
67+
);
6568

6669
cmakeFlags = [
6770
"-DKOMPUTE_OPT_USE_BUILT_IN_VULKAN_HEADER=OFF"

0 commit comments

Comments
 (0)