-
Notifications
You must be signed in to change notification settings - Fork 1
/
Singularity.ubuntu_tf
66 lines (50 loc) · 1.46 KB
/
Singularity.ubuntu_tf
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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
Bootstrap: docker
From: tensorflow/tensorflow:2.3.0-custom-op-gpu-ubuntu16
%environment
# use bash as default shell
SHELL=/bin/bash
export SHELL
export HDF5_USE_FILE_LOCKING='FALSE'
%setup
# runs on host
# the path to the image is $SINGULARITY_ROOTFS
%post
# post-setup script
# load environment variables
. /environment
# use bash as default shell
echo 'SHELL=/bin/bash' >> /environment
# make environment file executable
chmod +x /environment
# default mount paths
mkdir /scratch
# fix this issue: https://github.com/singularityware/singularity/issues/1182#issuecomment-381796545
touch /usr/bin/nvidia-smi
apt-get update && apt-get -y install locales
locale-gen en_US.UTF-8
apt-get install -y git wget
apt-get clean
# scikit, scipy
apt-get install -y python3-scipy python3-sklearn python3-sklearn-lib
pip install --upgrade pip
pip uninstall -y numpy
pip install numpy
#pip install -U scikit-learn
pip install Pillow
pip install matplotlib
pip install simpleitk
pip install comet_ml
pip install pandas
pip install tqdm
pip install h5py
pip install spektral
pip install -e git+https://github.com/marcoancona/DeepExplain.git#egg=deepexplain
pip install tf-explain
pip install tensorflow-addons
pip install imgaug
#pip install einops
%runscript
# executes with the singularity run command
# delete this section to use existing docker ENTRYPOINT command
%test
# test that script is a success