fix ai gpu
Some checks are pending
Flake.lock / build (push) Waiting to run

This commit is contained in:
Mohamed Chrayed 2024-08-28 17:25:22 +02:00
parent 6faee239c1
commit 4d110c5f62

View File

@ -1,8 +1,15 @@
{ ... }:
{ pkgs, ... }:
{
environment.systemPackages = [
pkgs.alpaca
];
environment.systemPackages =
(with pkgs; [
ollama-rocm
]);
services.ollama = {
enable = true;
acceleration = "rocm";
environmentVariables = {
HCC_AMDGPU_TARGET = "gfx1100";
HSA_OVERRIDE_GFX_VERSION="11.0.0";
};
};
}