tensorflow (without cuda)

This commit is contained in:
Danilo Reyes 2023-12-01 10:14:25 -06:00
parent fbd9822800
commit d9194fad76
2 changed files with 13 additions and 2 deletions

View File

@ -696,6 +696,12 @@ some applications use.
#+begin_src nix #+begin_src nix
environment = { environment = {
systemPackages = with pkgs; [ systemPackages = with pkgs; [
# needed for tensorflow
# linuxPackages.nvidia_x11
# cudatoolkit
# cudaPackages.cudnn
# cudaPackages.cutensor
wget wget
jellyfin-ffmpeg # coolest video converter! jellyfin-ffmpeg # coolest video converter!
mediainfo mediainfo
@ -779,6 +785,12 @@ environment = {
"\${XDG_DATA_HOME}/npm/bin" "\${XDG_DATA_HOME}/npm/bin"
"\${XDG_DATA_HOME}/pnpm" "\${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 #+end_src

View File

@ -20,8 +20,7 @@ in {
extraGroups = [ "render" ]; extraGroups = [ "render" ];
packages = (with pkgs; [ packages = (with pkgs; [
nodejs nodejs
python3 (python3.withPackages (ps: with ps; [ tensorflow ]))
python3Packages.tensorflowWithCuda
perl perl
(perlPackages.buildPerlPackage rec { (perlPackages.buildPerlPackage rec {
pname = "Image-ExifTool"; pname = "Image-ExifTool";