-
Notifications
You must be signed in to change notification settings - Fork 7
/
deic_init.sh
22 lines (19 loc) · 1.01 KB
/
deic_init.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
#!/usr/bin/env bash
sudo apt-get update && sudo apt-get install -y libgl1-mesa-dev
eval "$(conda shell.bash hook)"
# Check if dir /work/cenv/ssl4eo exists; create conda env if not
if [ -d "/work/project/cenv/ssl4eo" ]; then
export CONDA_ENVS_DIRS="/work/project/cenv/"
echo 'export CONDA_ENVS_DIRS="/work/project/cenv/"' >> ~/.bashrc
mamba env config vars set -n ssl4eo MMEARTH_DIR=/work/data/MMEARTH100K/ GEO_BENCH_DIR=/work/data/geobench
else
echo "Creating conda environment"
mamba env create --prefix /work/project/cenv/ssl4eo -f /work/data/env.yml
export CONDA_ENVS_DIRS="/work/project/cenv/"
echo 'export CONDA_ENVS_DIRS="/work/project/cenv/"' >> ~/.bashrc
# Set environment variables ensuring that env vars are always set when activating env
mamba env config vars set -n ssl4eo MMEARTH_DIR=/work/data/MMEARTH100K/ GEO_BENCH_DIR=/work/data/geobench
fi
mamba init
# Export new python kernel
/work/project/cenv/ssl4eo/bin/python -m ipykernel install --user --name ipy39 --display-name "SSL4EO"