shell.nix

# Helpful when fixing this recipe
#
# https://github.com/cdepillabout/course-v3/blob/6ecf90f640f327dc05b167f45fb7ba88bc265281/default.nix

let
  nixpkgs = import ./nix/nixpkgs_version.nix;
in
  nixpkgs.stdenv.mkDerivation rec {
    name = "fastai-build";
    env = nixpkgs.buildEnv { name=name; paths=buildInputs; };
    buildInputs = ( import ./nix/build.nix { inherit nixpkgs; });
    src = null;
    shellHook = ''
      SOURCE_DATE_EPOCH=$(date +%s)
      export PYTHONUSERBASE=$PWD/.local
      export USER_SITE=`python -c "import site; print(site.USER_SITE)"`
      export PYTHONPATH=$PYTHONPATH:$USER_SITE
      export PATH=$PATH:$PYTHONUSERBASE/bin

      jupyter nbextension install --py widgetsnbextension --user > /dev/null 2>&1
      jupyter nbextension enable widgetsnbextension --user --py > /dev/null 2>&1
      pip install jupyter_contrib_nbextensions --user > /dev/null 2>&1
      jupyter contrib nbextension install --user > /dev/null 2>&1
      jupyter nbextension enable spellchecker/main > /dev/null 2>&1

      export LD_PRELOAD="/usr/lib/x86_64-linux-gnu/libcuda.so.418.39 /usr/lib/nvidia-418/libnvidia-fatbinaryloader.so.418.39 /usr/lib/nvidia-418/libnvidia-ptxjitcompiler.so.418.39 /usr/lib/nvidia-418/libnvidia-ml.so.418.39"

      export ZONE="us-east1-c"
      export INSTANCE_NAME="my-fastai-instance"
      export INSTANCE_TYPE="n1-highmem-8"

    '';
}