From d9194fad76ef0e2da05259e505b76465110e0954 Mon Sep 17 00:00:00 2001 From: Danilo Reyes Date: Fri, 1 Dec 2023 10:14:25 -0600 Subject: [PATCH] tensorflow (without cuda) --- server/configuration.org | 12 ++++++++++++ server/servers.nix | 3 +-- 2 files changed, 13 insertions(+), 2 deletions(-) diff --git a/server/configuration.org b/server/configuration.org index e9a424f..001e876 100644 --- a/server/configuration.org +++ b/server/configuration.org @@ -696,6 +696,12 @@ some applications use. #+begin_src nix environment = { systemPackages = with pkgs; [ + # needed for tensorflow + # linuxPackages.nvidia_x11 + # cudatoolkit + # cudaPackages.cudnn + # cudaPackages.cutensor + wget jellyfin-ffmpeg # coolest video converter! mediainfo @@ -779,6 +785,12 @@ environment = { "\${XDG_DATA_HOME}/npm/bin" "\${XDG_DATA_HOME}/pnpm" ]; + + # needed for tensorflow + # CUDA_PATH = "${pkgs.cudatoolkit}"; + # # LD_LIBRARY_PATH = "${pkgs.linuxPackages.nvidia_x11}/lib:${pkgs.ncurses5}/lib"; + # EXTRA_LDFLAGS = "-L/lib -L${pkgs.linuxPackages.nvidia_x11}/lib"; + # EXTRA_CCFLAGS = "-I/usr/include"; }; }; #+end_src diff --git a/server/servers.nix b/server/servers.nix index f99f61e..b4d1b99 100644 --- a/server/servers.nix +++ b/server/servers.nix @@ -20,8 +20,7 @@ in { extraGroups = [ "render" ]; packages = (with pkgs; [ nodejs - python3 - python3Packages.tensorflowWithCuda + (python3.withPackages (ps: with ps; [ tensorflow ])) perl (perlPackages.buildPerlPackage rec { pname = "Image-ExifTool";