-
Notifications
You must be signed in to change notification settings - Fork 0
/
Singularity.caffe-opa
55 lines (46 loc) · 1.8 KB
/
Singularity.caffe-opa
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
BootStrap: yum
OSVersion: 7
MirrorURL: http://mirror.centos.org/centos-7/7.6.1810/os/x86_64/
Include: yum
%help
Containerized Intel Caffe and Intel Omni-Path Architecture
%labels
Maintainer Ray Marc Marcellones
Version v0.1
%post
yum -y install epel-release
yum -y update
yum -y groupinstall "Development Tools"
yum -y install wget cmake git
yum -y install protobuf-devel protobuf-compiler boost-devel
yum -y install snappy-devel opencv-devel atlas-devel
yum -y install gflags-devel glog-devel lmdb-devel leveldb-devel hdf5-devel
# Intel Omni-Path Architecture
yum -y install perl atlas libpsm2 infinipath-psm libibverbs qperf pciutils tcl tcsh
yum -y install expect sysfsutils librdmacm libibcm perftest rdma bc
yum -y install elfutils-libelf-developenssh-clients openssh-server
yum -y install libstdc++-devel gcc-gfortran rpm-buildx
yum -y install compat-rdma-devel libibmad libibumad ibacm-devel
yum -y install libibumad-devel libibumad-static libuuid-devel
yum -y install irqbalance openssl openssl-devel
# Intel Omni-Path Architecture
mkdir /opt/tempdir
cd /opt/tempdir
wget -O OPA https://downloadmirror.intel.com/28721/eng/IntelOPA-Basic.RHEL76-x86_64.10.9.2.0.9.tgz
tar -xvf OPA
cd IntelOPA-Basic.RHEL76-x86_64.10.9.2.0.9
./INSTALL --user-space -n
rm -rf /opt/tempdir
# CAFFE
cd /opt
git clone https://github.com/intel/caffe.git intel-caffe
cd intel-caffe
cp Makefile.config.example Makefile.config
NUM_THREADS=$(($(grep 'core id' /proc/cpuinfo | sort -u | wc -l)*2))
make -j $NUM_THREADS
%environment
source /opt/intel-caffe/external/mlsl/l_mlsl_2018.1.005/intel64/bin/mlslvars.sh
CAFFE_ROOT=/opt/intel-caffe/build/tools/caffe
export CAFFE_ROOT
%runscript
exec $CAFFE_ROOT