blob: 48cb5aef72fddc7847bd98d9a32fb3095d88b16c (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
|
{ config, pkgs, lib, ... }:
{
boot = {
initrd = {
kernelModules = [ "amdgpu" ];
};
};
environment.systemPackages = with pkgs; [
rocmPackages.rocminfo
rocmPackages.rocm-smi
rocmPackages.hipcc
];
hardware = {
graphics = {
enable = true;
extraPackages = with pkgs; [
rocmPackages.clr
];
};
};
nixpkgs.config = {
allowUnfree = lib.mkForce true;
rocmSupport = true;
rocmTargets = [ "gfx1031" ];
};
services = {
ollama = {
environmentVariables = {
HCC_AMDGPU_TARGET = "gfx1031";
};
rocmOverrideGfx = "10.3.0";
};
xserver = {
enable = true;
videoDrivers = [
"amdgpu"
];
};
};
}
|
