Skip to content

Commit

Permalink
export additional wandb variables to avoid clogging home
Browse files Browse the repository at this point in the history
  • Loading branch information
svandenhaute committed Jun 7, 2023
1 parent dfb3c0a commit 8a0ae38
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion configs/lumi.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,8 @@
# define provider for default executor (HTEX)
# each of the workers in this executor is single-core;
# they do basic processing stuff (reading/writing data/models, ... )
worker_init = 'export SINGULARITYENV_WANDB_CACHE_DIR="$(pwd)"\n'
worker_init = 'export SINGULARITYENV_WANDB_CACHE_DIR="$(pwd)"\n'
worker_init += 'export SINGULARITYENV_WANDB_DATA_DIR="$(pwd)"\n'
provider = SlurmProviderVSC( # one block == one slurm job to submit
cluster='lumi',
partition='small',
Expand All @@ -80,6 +81,8 @@
worker_init += 'export SINGULARITYENV_ROCM_PATH="/opt/rocm"\n'
worker_init += 'export SINGULARITYENV_ROCBLAS_TENSILE_LIBPATH="/opt/rocm/lib/rocblas/library/"\n'
worker_init += 'export SINGULARITYENV_XTPE_LINK_TYPE="dynamic\n"'
worker_init += 'export SINGULARITYENV_WANDB_CACHE_DIR="$(pwd)"\n'
worker_init += 'export SINGULARITYENV_WANDB_DATA_DIR="$(pwd)"\n'
provider = SlurmProviderVSC(
cluster='lumi',
partition='eap',
Expand Down

0 comments on commit 8a0ae38

Please sign in to comment.