

deps: add cuda packages to flake.nix required by onnxruntime-gpu
@a6e2024ee5ae32037ff3d51fb9b239df7bcc3563
--- flake.nix
+++ flake.nix
... | ... | @@ -38,8 +38,15 @@ |
38 | 38 |
# NOTE: sometimes it still doesn't work but rebooting the system fixes it |
39 | 39 |
LD_LIBRARY_PATH = "/run/opengl-driver/lib:${ |
40 | 40 |
pkgs.lib.makeLibraryPath [ |
41 |
+ # Needed for `faster-whisper` |
|
41 | 42 |
pkgs.cudaPackages_12.cudnn |
42 | 43 |
pkgs.cudaPackages_12.libcublas |
44 |
+ # The 4 cuda packages below are needed for `onnxruntime-gpu` |
|
45 |
+ pkgs.cudaPackages_12.libcurand |
|
46 |
+ pkgs.cudaPackages_12.libcufft |
|
47 |
+ pkgs.cudaPackages_12.cuda_cudart |
|
48 |
+ pkgs.cudaPackages_12.cuda_nvrtc |
|
49 |
+ |
|
43 | 50 |
pkgs.zlib |
44 | 51 |
pkgs.stdenv.cc.cc |
45 | 52 |
pkgs.openssl |
Add a comment
Delete comment
Once you delete this comment, you won't be able to recover it. Are you sure you want to delete this comment?