File tree 4 files changed +51
-7
lines changed
4 files changed +51
-7
lines changed Original file line number Diff line number Diff line change @@ -2,7 +2,7 @@ defaults:
2
2
- model : resnet
3
3
- loss : cross_entropy
4
4
- optim : sgd
5
- - scheduler : step
5
+ - scheduler : cosine
6
6
- dataset : imagenet
7
7
- pipeline : classifier
8
8
- hydra/job_logging : colorlog
Original file line number Diff line number Diff line change @@ -18,5 +18,5 @@ services:
18
18
command : ['sleep', 'infinity']
19
19
volumes :
20
20
- .:/code
21
- - /data1 /data:/data
22
- - /data2/data /train_log:/outputs
21
+ - /DATA/disk2/xlm /data:/data
22
+ - /DATA/disk3 /train_log:/outputs
Original file line number Diff line number Diff line change @@ -9,9 +9,9 @@ RUN pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
9
9
RUN pip install --no-cache-dir -r /tmp/requirements.txt
10
10
RUN CC="cc -mavx2" pip install -U --force-reinstall pillow-simd
11
11
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
16
16
17
17
USER ${USER_NAME}
Original file line number Diff line number Diff line change
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"
You can’t perform that action at this time.
0 commit comments