|
a |
|
b/tasks_adapt/script.sh.template |
|
|
1 |
hostname |
|
|
2 |
nvidia-smi |
|
|
3 |
# Container starting directory determined by settings in Docker image |
|
|
4 |
# For NVIDIA NGC containers this is normally /workspace |
|
|
5 |
echo Container starts in the directory: |
|
|
6 |
pwd |
|
|
7 |
# Install required packages |
|
|
8 |
# Set environment variable to install packages to world-writable location inside container |
|
|
9 |
export PYTHONUSERBASE=/workspace/.local |
|
|
10 |
# Escaping PYTHONUSERBASE in the next line so that it is expanded inside docker and take the value set inside the container |
|
|
11 |
mkdir -p $PYTHONUSERBASE/bin |
|
|
12 |
export PATH=$PATH:$PYTHONUSERBASE/bin |
|
|
13 |
echo PATH inside container |
|
|
14 |
echo PATH=$PATH |
|
|
15 |
pip install -U -q --no-cache-dir --user Braindecode==0.4.85 |
|
|
16 |
|
|
|
17 |
# change directory to correct location if required |
|
|
18 |
cd /home/users/ntu/kzhang01/hbm/eeg-adapt |
|
|
19 |
echo Changed working directory to |
|
|
20 |
pwd |
|
|
21 |
|
|
|
22 |
datapath=/home/users/ntu/kzhang01/scratch/KU_mi_smt.h5 |
|
|
23 |
modelpath=/home/users/ntu/kzhang01/hbm/eeg-adapt/result_base/best |
|
|
24 |
logpath=/home/users/ntu/kzhang01/hbm/eeg-adapt/tasks_adapt{0} |
|
|
25 |
# Fix file locking disabled on this file system (/scratch) when reading hdf5 |
|
|
26 |
export HDF5_USE_FILE_LOCKING='FALSE' |
|
|
27 |
|