-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathDockerfile
48 lines (40 loc) · 1.01 KB
/
Dockerfile
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
FROM docker.io/nvidia/cuda:11.3.1-cudnn8-runtime-ubuntu18.04
ENV \
LANG=C.UTF-8 \
JAVA_HOME=/usr/lib/jvm/java-11-openjdk-amd64 \
LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
ENV DEBIAN_FRONTEND=noninteractive
RUN apt-get update &&\
apt-get install -y --no-install-recommends \
wget \
zip \
unzip \
ca-certificates \
build-essential \
curl \
libcurl4-openssl-dev \
libssl-dev \
libgomp1 \
openjdk-11-jdk \
ca-certificates-java
RUN wget -q -O /tmp/sbt-1.5.5.tgz https://github.com/sbt/sbt/releases/download/v1.5.5/sbt-1.5.5.tgz &&\
tar -zxf /tmp/sbt-1.5.5.tgz --strip=1 -C /usr
WORKDIR /jar
ADD build.sbt ./
ADD src ./src/
ADD model ./model/
ADD project/build.properties ./project/
ADD project/plugins.sbt ./project/
RUN sbt clean assembly
ENTRYPOINT exec java $JVM_OPTS \
-Dport=8080 \
-Dbatch=16 \
-Dtimeout=10000 \
-Ddim=100 \
-Dsample=10000 \
-DnpyFile=./model/10000-100.npy \
-DtakeSpinCountDelay=5 \
-Xmx8G \
-Xms8G \
-DLOG_LEVEL=${LOG_LEVEL} \
-jar ./target/scala-2.12/flasma.jar