wandb
51 строка · 1.4 Кб
1FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu20.04
2
3ARG DEBIAN_FRONTEND=noninteractive
4
5ARG PYTHON_VERSION=3.9
6ARG GIT_BRANCH=master
7ARG TOX_VERSION=4.5.1
8ARG UTC_DATE
9
10# todo: enable docker image layer caching on circleci
11RUN apt-get update && apt-get install -y --no-install-recommends \
12vim \
13curl \
14ca-certificates \
15sudo \
16git \
17python${PYTHON_VERSION} \
18python3-pip \
19python${PYTHON_VERSION}-dev \
20build-essential \
21libsndfile1 \
22ffmpeg \
23&& apt-get clean && rm -rf /var/lib/apt/lists/* \
24&& ln -s /usr/bin/python3 /usr/bin/python
25
26RUN mkdir /wandb
27WORKDIR /wandb
28
29RUN adduser --disabled-password --gecos '' --shell /bin/bash sdk \
30&& adduser sdk sudo\
31&& chown -R sdk:sdk /wandb
32RUN echo "sdk ALL=(ALL) NOPASSWD:ALL" > /etc/sudoers.d/90-sdk
33USER sdk
34ENV HOME=/home/sdk
35RUN chmod 777 /home/sdk
36
37# clone wandb/wandb repository and install wandb sdk
38RUN git clone https://github.com/wandb/wandb.git /wandb/wandb \
39&& cd /wandb/wandb \
40&& git checkout ${GIT_BRANCH} \
41&& pip install --upgrade pip \
42&& pip install tox==${TOX_VERSION} --no-cache-dir
43
44RUN PATH=/home/sdk/.local/bin:$PATH
45
46WORKDIR /wandb/wandb
47# use torch wheels with CUDA 11.8 support
48RUN sed -i -e 's/whl\/cpu/whl\/cu118/g' tox.ini
49ENV DATE=$UTC_DATE
50#CMD ["tail", "-f", "/dev/null"]
51CMD ["python", "-m", "tox", "-v", "-e", "standalone-gpu-py39,pod-store"]
52