32 lines
886 B
Nix
32 lines
886 B
Nix
|
# Need nvidia proprietary drivers to work
|
||
|
|
{ pkgs, nixpkgs, config, lib, ... }:
|
||
|
|
{
|
||
|
|
config = lib.mkIf config.frogeye.cuda {
|
||
|
|
nix.settings = {
|
||
|
|
substituters = [
|
||
|
|
"https://cuda-maintainers.cachix.org"
|
||
|
|
];
|
||
|
|
trusted-public-keys = [
|
||
|
|
"cuda-maintainers.cachix.org-1:0dq3bujKpuEPMCX6U4WylrUDZ9JyUG0VpVZa7CNfq5E="
|
||
|
|
];
|
||
|
|
};
|
||
|
|
nixpkgs.overlays = [
|
||
|
|
(self: super:
|
||
|
|
let
|
||
|
|
pkgs_uf = import nixpkgs {
|
||
|
|
inherit (super) system;
|
||
|
|
config.allowUnfree = true;
|
||
|
|
};
|
||
|
|
in
|
||
|
|
{
|
||
|
|
ctranslate2 = (pkgs_uf.ctranslate2.override {
|
||
|
|
withCUDA = true;
|
||
|
|
withCuDNN = true;
|
||
|
|
stdenv = pkgs.gcc12Stdenv; # 24.11 Remove, as nvcc and gcc versions align again
|
||
|
|
# Takes more than 16 GiB RAM to compile with the above, but fine if not
|
||
|
|
});
|
||
|
|
})
|
||
|
|
];
|
||
|
|
};
|
||
|
|
}
|