Mortal
This commit is contained in:
42
Dockerfile
Normal file
42
Dockerfile
Normal file
@@ -0,0 +1,42 @@
|
||||
# syntax=docker/dockerfile:1.4-labs
|
||||
|
||||
FROM archlinux:base-devel as libriichi_build
|
||||
|
||||
RUN <<EOF
|
||||
pacman -Syu --noconfirm --needed rust python
|
||||
pacman -Scc
|
||||
EOF
|
||||
|
||||
WORKDIR /
|
||||
COPY Cargo.toml Cargo.lock .
|
||||
COPY libriichi libriichi
|
||||
|
||||
RUN cargo build -p libriichi --lib --release
|
||||
|
||||
# -----
|
||||
FROM archlinux:base
|
||||
|
||||
RUN <<EOF
|
||||
pacman -Syu --noconfirm --needed python python-pytorch python-toml python-tqdm tensorboard
|
||||
pacman -Scc
|
||||
EOF
|
||||
|
||||
WORKDIR /mortal
|
||||
COPY mortal .
|
||||
COPY --from=libriichi_build /target/release/libriichi.so .
|
||||
|
||||
ENV MORTAL_CFG config.toml
|
||||
COPY <<'EOF' config.toml
|
||||
[control]
|
||||
state_file = '/mnt/mortal.pth'
|
||||
|
||||
[resnet]
|
||||
conv_channels = 192
|
||||
num_blocks = 40
|
||||
enable_bn = true
|
||||
bn_momentum = 0.99
|
||||
EOF
|
||||
|
||||
VOLUME /mnt
|
||||
|
||||
ENTRYPOINT ["python", "mortal.py"]
|
||||
Reference in New Issue
Block a user