ollama rocm gpu fix
This commit is contained in:
@ -20,5 +20,9 @@ in {
|
||||
host = "[::]";
|
||||
openFirewall = true;
|
||||
};
|
||||
nixpkgs.config = {
|
||||
rocmSupport = config.services.xserver.videoDrivers == ["amdgpu"];
|
||||
cudaSupport = config.services.xserver.videoDrivers == ["nvidia"];
|
||||
};
|
||||
};
|
||||
}
|
||||
|
@ -48,4 +48,10 @@
|
||||
podman.enable = true;
|
||||
virtualisation.enable = true;
|
||||
};
|
||||
services.ollama = {
|
||||
environmentVariables = {
|
||||
HCC_AMDGPU_TARGET = "gfx1100";
|
||||
};
|
||||
rocmOverrideGfx = "11.0.0";
|
||||
};
|
||||
}
|
||||
|
Reference in New Issue
Block a user