Skip to content

Commit 28d6dff

Browse files
committedJul 14, 2021
update
1 parent c6718f1 commit 28d6dff

File tree

4 files changed

+51
-7
lines changed

4 files changed

+51
-7
lines changed
 

‎conf/config.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@ defaults:
22
- model: resnet
33
- loss: cross_entropy
44
- optim: sgd
5-
- scheduler: step
5+
- scheduler: cosine
66
- dataset: imagenet
77
- pipeline: classifier
88
- hydra/job_logging: colorlog

‎docker-compose.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,5 +18,5 @@ services:
1818
command: ['sleep', 'infinity']
1919
volumes:
2020
- .:/code
21-
- /data1/data:/data
22-
- /data2/data/train_log:/outputs
21+
- /DATA/disk2/xlm/data:/data
22+
- /DATA/disk3/train_log:/outputs

‎docker/Dockerfile.local

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -9,9 +9,9 @@ RUN pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
99
RUN pip install --no-cache-dir -r /tmp/requirements.txt
1010
RUN CC="cc -mavx2" pip install -U --force-reinstall pillow-simd
1111

12-
RUN addgroup --gid ${GROUP_ID} ${USER_NAME}
13-
RUN adduser --disabled-password --gecos '' --uid $USER_ID --gid $GROUP_ID ${USER_NAME}
14-
15-
RUN cp /root/.zshrc /home/${USER_NAME}/
12+
RUN if [ $USER_NAME != "root" ] ; \
13+
then addgroup --gid ${GROUP_ID} ${USER_NAME} \
14+
&& adduser --disabled-password --gecos '' --uid $USER_ID --gid $GROUP_ID ${USER_NAME} \
15+
&& cp /root/.zshrc /home/${USER_NAME}/ ; fi
1616

1717
USER ${USER_NAME}

‎docker/Dockerfile.pytorch-1.6.0

Lines changed: 44 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,44 @@
1+
# build deepbase/pytorch:1.6.0
2+
FROM nvidia/cuda:10.2-cudnn8-devel-ubuntu18.04
3+
4+
ENV PYTORCH_VERSION=1.6.0
5+
ENV TORCHVISION_VERSION=0.7.0
6+
7+
ARG python=3.8
8+
ENV PYTHON_VERSION=${python}
9+
10+
# Set default shell to /bin/bash
11+
SHELL ["/bin/bash", "-cu"]
12+
13+
ENV DEBIAN_FRONTEND=noninteractive
14+
RUN apt-get update && apt-get install -y --allow-downgrades --allow-change-held-packages --no-install-recommends \
15+
build-essential \
16+
cmake \
17+
g++-7 \
18+
git \
19+
curl \
20+
vim \
21+
wget \
22+
ca-certificates \
23+
libjpeg-dev \
24+
libpng-dev \
25+
librdmacm1 \
26+
libibverbs1 \
27+
ibverbs-providers \
28+
zsh \
29+
tzdata
30+
31+
RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O /tmp/install-conda.sh \
32+
&& chmod +x /tmp/install-conda.sh \
33+
&& /tmp/install-conda.sh -b -f -p /usr/local \
34+
&& rm -f /tmp/install-conda.sh \
35+
&& conda install -y python=${PYTHON_VERSION} \
36+
&& conda clean -y --all
37+
38+
# Install PyTorch
39+
RUN pip install --no-cache-dir torch==${PYTORCH_VERSION} torchvision==${TORCHVISION_VERSION}
40+
41+
ENV ZSH /usr/local/.oh-my-zsh
42+
RUN curl https://raw.githubusercontent.com/robbyrussell/oh-my-zsh/master/tools/install.sh | bash -s -- --unattended
43+
44+
WORKDIR "/code"

0 commit comments

Comments
 (0)
Please sign in to comment.