RepVap/default.nix at main · Corallus-Caninus/RepVap · GitHub

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
{ pkgs ? import <nixpkgs> { config.allowUnfree = true; config.nvidia.acceptLicense = true; } }:
#TODO: how can we programatically install the virtual environment for the python dependencies?
pkgs.mkShell {
  buildInputs = [
    pkgs.python39
    pkgs.python39Packages.pip
    pkgs.pkg-config
    pkgs.python39Packages.tkinter
#    pkgs.cudaPackages_11.cudatoolkit
#    pkgs.linuxPackages.nvidia_x11_legacy470
#    pkgs.glibc
#    pkgs.glib
#    pkgs.gcc12
#    pkgs.gcc-unwrapped
#    pkgs.ninja
#    pkgs.libxcrypt
#    pkgs.python3Packages.pytorchWithCuda
#    pkgs.python3Packages.transformers
#    pkgs.python3Packages.datasets
#    pkgs.python3Packages.peft
#    pkgs.python3Packages.pip
  ];
#  shellHook = ''
#    echo "You are now using a NIX environment"
#    export CUDA_PATH=${pkgs.cudaPackages_11.cudatoolkit}
##    export LD_LIBRARY_PATH=${pkgs.stdenv.cc.cc.lib}/lib
#    export LD_LIBRARY_PATH=${pkgs.stdenv.cc.cc.lib}/lib:${pkgs.gcc12}/lib
##    export LD_LIBRARY_PATH=${pkgs.gcc11}/lib
#    export EXTRA_LD_FLAGS="-L\/lib -L${pkgs.linuxPackages.nvidia_x11_legacy470}\/lib"
##    export LD_LIBRARY_PATH=${pkgs.linuxPackages.nvidia_x11_legacy470}/lib:${pkgs.cudaPackages.cudatoolkit}/lib64:$LD_LIBRARY_PATH
#    export LD_LIBRARY_PATH=${pkgs.linuxPackages.nvidia_x11_legacy470}/lib:$LD_LIBRARY_PATH
#    export LD_LIBRARY_PATH=${pkgs.python39}/lib:$LD_LIBRARY_PATH
#    export LD_LIBRARY_PATH=$CUDA_PATH/lib:$LD_LIBRARY_PATH
#export PKG_CONFIG_PATH="${pkgs.python39}/lib/pkgconfig:$PKG_CONFIG_PATH"
#    alias gcc="${pkgs.gcc12}/bin/gcc"
#  '';
}