Skip to content

Commit 9886bd1

Browse files
rlanrlan
committed
Fix #97 
1 parent eecc22e commit 9886bd1

26 files changed

+270
-3
lines changed

Diff for: ‎README.md

+3-3
Original file line numberDiff line numberDiff line change
@@ -52,9 +52,9 @@ Images of Pytorch version 1.5 and higher include [Pytorch Lightning](https://git
5252

5353
| Tag (OS-based python) | Comment | Dockerfile | Info |
5454
| --------------------- | ------- | ---------- | ---- |
55-
| `pytorch1.10.0` | CPU-only | [Dockerfile](docker/pytorch1.10.0/Dockerfile) | ![](https://img.shields.io/docker/image-size/wqael/notebooks/pytorch1.10.0) |
56-
| `pytorch1.10.0-cuda11.3.1` | Nvidia Driver >= 465.19.01 (Linux) 465.89 (Windows) | [Dockerfile](docker/pytorch1.10.0-cuda11.3.1/Dockerfile) | ![](https://img.shields.io/docker/image-size/wqael/notebooks/pytorch1.10.0-cuda11.3.1) |
57-
| `pytorch1.10.0-cuda10.2` | Nvidia Driver >= 440.33 (Linux) 441.22 (Windows) | [Dockerfile](docker/pytorch1.10.0-cuda10.2/Dockerfile) | ![](https://img.shields.io/docker/image-size/wqael/notebooks/pytorch1.10.0-cuda10.2) |
55+
| `pytorch1.11.0` | CPU-only | [Dockerfile](docker/pytorch1.11.0/Dockerfile) | ![](https://img.shields.io/docker/image-size/wqael/notebooks/pytorch1.11.0) |
56+
| `pytorch1.11.0-cuda11.3.1` | Nvidia Driver >= 465.19.01 (Linux) 465.89 (Windows) | [Dockerfile](docker/pytorch1.11.0-cuda11.3.1/Dockerfile) | ![](https://img.shields.io/docker/image-size/wqael/notebooks/pytorch1.11.0-cuda11.3.1) |
57+
| `pytorch1.11.0-cuda10.2` | Nvidia Driver >= 440.33 (Linux) 441.22 (Windows) | [Dockerfile](docker/pytorch1.11.0-cuda10.2/Dockerfile) | ![](https://img.shields.io/docker/image-size/wqael/notebooks/pytorch1.11.0-cuda10.2) |
5858
| `pytorch1.9.1-cuda11.1.1` | Nvidia Driver >= 455.32 (Linux) 456.81 (Windows) | [Dockerfile](docker/pytorch1.9.1-cuda11.1.1/Dockerfile) | ![](https://img.shields.io/docker/image-size/wqael/notebooks/pytorch1.9.1-cuda11.1.1) |
5959
| `pytorch1.7.1-cuda11` | Nvidia Driver >= 450.36.06 | [Dockerfile](docker/pytorch1.7.1-cuda11/Dockerfile) | ![](https://img.shields.io/docker/image-size/wqael/notebooks/pytorch1.7.1-cuda11) |
6060
| `pytorch1.7.1-cuda101` | Nvidia Driver >= 418.xx | [Dockerfile](docker/pytorch1.7.1-cuda101/Dockerfile) | ![](https://img.shields.io/docker/image-size/wqael/notebooks/pytorch1.7.1-cuda101) |

Diff for: ‎deprecated/deprecated.md

+3
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,9 @@ Caffe2 APIs are being [deprecated](https://caffe2.ai/blog/2018/05/02/Caffe2_PyTo
6868

6969
| Tag (OS-based python) | Comment | Dockerfile | Info |
7070
| --------------------- | ------- | ---------- | ---- |
71+
| `pytorch1.10.0` | CPU-only | [Dockerfile](pytorch1.10.0/Dockerfile) | ![](https://img.shields.io/docker/image-size/wqael/notebooks/pytorch1.10.0) |
72+
| `pytorch1.10.0-cuda11.3.1` | Nvidia Driver >= 465.19.01 (Linux) 465.89 (Windows) | [Dockerfile](pytorch1.10.0-cuda11.3.1/Dockerfile) | ![](https://img.shields.io/docker/image-size/wqael/notebooks/pytorch1.10.0-cuda11.3.1) |
73+
| `pytorch1.10.0-cuda10.2` | Nvidia Driver >= 440.33 (Linux) 441.22 (Windows) | [Dockerfile](pytorch1.10.0-cuda10.2/Dockerfile) | ![](https://img.shields.io/docker/image-size/wqael/notebooks/pytorch1.10.0-cuda10.2) |
7174
| `pytorch1.9.0` | CPU-only | [Dockerfile](pytorch1.9.0/Dockerfile) | |
7275
| `pytorch1.9.0-cuda11.1.1` | Nvidia Driver >= 455.32 | [Dockerfile](pytorch1.9.0-cuda11.1.1/Dockerfile) | |
7376
| `pytorch1.9.0-cuda10.2` | Nvidia Driver >= 440.33 | [Dockerfile](pytorch1.9.0-cuda10.2/Dockerfile) | |

Diff for: ‎docker/pytorch1.10.0-cuda10.2/Dockerfile renamed to ‎deprecated/pytorch1.10.0-cuda10.2/Dockerfile

File renamed without changes.

Diff for: ‎docker/pytorch1.10.0-cuda10.2/hooks/build renamed to ‎deprecated/pytorch1.10.0-cuda10.2/hooks/build

File renamed without changes.

Diff for: ‎docker/pytorch1.10.0-cuda10.2/jupyter_notebook_config.py renamed to ‎deprecated/pytorch1.10.0-cuda10.2/jupyter_notebook_config.py

File renamed without changes.

Diff for: ‎docker/pytorch1.10.0-cuda10.2/run_jupyter.sh renamed to ‎deprecated/pytorch1.10.0-cuda10.2/run_jupyter.sh

File renamed without changes.

Diff for: ‎docker/pytorch1.10.0-cuda11.3.1/Dockerfile renamed to ‎deprecated/pytorch1.10.0-cuda11.3.1/Dockerfile

File renamed without changes.

Diff for: ‎docker/pytorch1.10.0-cuda11.3.1/hooks/build renamed to ‎deprecated/pytorch1.10.0-cuda11.3.1/hooks/build

File renamed without changes.

Diff for: ‎docker/pytorch1.10.0-cuda11.3.1/jupyter_notebook_config.py renamed to ‎deprecated/pytorch1.10.0-cuda11.3.1/jupyter_notebook_config.py

File renamed without changes.

Diff for: ‎docker/pytorch1.10.0-cuda11.3.1/run_jupyter.sh renamed to ‎deprecated/pytorch1.10.0-cuda11.3.1/run_jupyter.sh

File renamed without changes.

Diff for: ‎docker/pytorch1.10.0/Dockerfile renamed to ‎deprecated/pytorch1.10.0/Dockerfile

File renamed without changes.

Diff for: ‎docker/pytorch1.10.0/hooks/build renamed to ‎deprecated/pytorch1.10.0/hooks/build

File renamed without changes.

Diff for: ‎docker/pytorch1.10.0/jupyter_notebook_config.py renamed to ‎deprecated/pytorch1.10.0/jupyter_notebook_config.py

File renamed without changes.

Diff for: ‎docker/pytorch1.10.0/run_jupyter.sh renamed to ‎deprecated/pytorch1.10.0/run_jupyter.sh

File renamed without changes.

Diff for: ‎docker/pytorch1.11.0-cuda10.2/Dockerfile

+63
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,63 @@
1+
FROM wqael/docker:py3.8-cuda10.2
2+
3+
LABEL description="JupyterLab / PyTorch 1.11.0 / Python 3.8.12 / nvidia/cuda:10.2-cudnn8-runtime-ubuntu18.04" \
4+
maintainer="https://github.com/rlan/notebooks"
5+
6+
7+
# Build-time metadata as defined at http://label-schema.org
8+
ARG BUILD_DATE
9+
ARG VCS_REF
10+
ARG VERSION
11+
LABEL org.label-schema.build-date=$BUILD_DATE \
12+
org.label-schema.vcs-ref=$VCS_REF \
13+
org.label-schema.vcs-url="https://github.com/rlan/notebooks" \
14+
org.label-schema.version=$VERSION \
15+
org.label-schema.schema-version="1.0"
16+
17+
ARG PYTHON=python
18+
ARG PIP=pip
19+
# Skips tzdata configuration
20+
ARG DEBIAN_FRONTEND=noninteractive
21+
22+
RUN apt-get -qq update \
23+
&& apt-get -qq install -y --no-install-recommends \
24+
python3-pandas \
25+
&& apt-get -qq clean \
26+
&& rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*
27+
28+
RUN ${PIP} --no-cache-dir install -q --upgrade pip setuptools wheel
29+
30+
# Install pytorch
31+
RUN ${PIP} --no-cache-dir install -q --upgrade \
32+
torch torchvision torchaudio
33+
34+
RUN ${PIP} --no-cache-dir install -q --upgrade \
35+
jupyterlab \
36+
matplotlib \
37+
nltk \
38+
opencv-python-headless \
39+
pillow \
40+
pytorch-lightning \
41+
scikit-learn \
42+
scikit-image \
43+
seaborn \
44+
tensorboard \
45+
six
46+
47+
# Set up our notebook config.
48+
COPY jupyter_notebook_config.py /root/.jupyter/
49+
50+
# Jupyter has issues with being run directly:
51+
# https://github.com/ipython/ipython/issues/7062
52+
# We just add a little wrapper script.
53+
COPY run_jupyter.sh /
54+
55+
# IPython
56+
EXPOSE 8888
57+
58+
# Tensorboard
59+
EXPOSE 6006
60+
61+
WORKDIR /notebooks
62+
63+
CMD ["/run_jupyter.sh", "--allow-root"]

Diff for: ‎docker/pytorch1.11.0-cuda10.2/hooks/build

+5
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
#!/bin/bash
2+
# $IMAGE_NAME var is injected into the build so the tag is correct.
3+
docker build --build-arg VCS_REF=`git rev-parse --short HEAD` \
4+
--build-arg BUILD_DATE=`date -u +”%Y-%m-%dT%H:%M:%SZ”` \
5+
-t $IMAGE_NAME .

Diff for: ‎docker/pytorch1.11.0-cuda10.2/jupyter_notebook_config.py

+18
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
import os
2+
from IPython.lib import passwd
3+
4+
c = c # pylint:disable=undefined-variable
5+
c.NotebookApp.ip = '0.0.0.0'
6+
c.NotebookApp.port = int(os.getenv('PORT', 8888))
7+
c.NotebookApp.open_browser = False
8+
9+
# sets a password if PASSWORD is set in the environment
10+
if 'PASSWORD' in os.environ:
11+
password = os.environ['PASSWORD']
12+
if password:
13+
c.NotebookApp.password = passwd(password)
14+
else:
15+
c.NotebookApp.password = ''
16+
c.NotebookApp.token = ''
17+
del os.environ['PASSWORD']
18+

Diff for: ‎docker/pytorch1.11.0-cuda10.2/run_jupyter.sh

+2
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
#!/bin/bash
2+
jupyter notebook "$@"

Diff for: ‎docker/pytorch1.11.0-cuda11.3.1/Dockerfile

+63
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,63 @@
1+
FROM wqael/docker:py3.8-cuda11.3.1
2+
3+
LABEL description="JupyterLab / PyTorch 1.11.0 / Python 3.8.10 / nvidia/cuda:11.3.1-cudnn8-runtime-ubuntu20.04" \
4+
maintainer="https://github.com/rlan/notebooks"
5+
6+
7+
# Build-time metadata as defined at http://label-schema.org
8+
ARG BUILD_DATE
9+
ARG VCS_REF
10+
ARG VERSION
11+
LABEL org.label-schema.build-date=$BUILD_DATE \
12+
org.label-schema.vcs-ref=$VCS_REF \
13+
org.label-schema.vcs-url="https://github.com/rlan/notebooks" \
14+
org.label-schema.version=$VERSION \
15+
org.label-schema.schema-version="1.0"
16+
17+
ARG PYTHON=python3
18+
ARG PIP=pip3
19+
# Skips tzdata configuration
20+
ARG DEBIAN_FRONTEND=noninteractive
21+
22+
RUN apt-get -qq update \
23+
&& apt-get -qq install -y --no-install-recommends \
24+
python3-pandas \
25+
&& apt-get -qq clean \
26+
&& rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*
27+
28+
RUN ${PIP} --no-cache-dir install -q --upgrade pip setuptools wheel
29+
30+
# Install pytorch
31+
RUN ${PIP} --no-cache-dir install -q --upgrade \
32+
torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
33+
34+
RUN ${PIP} --no-cache-dir install -q --upgrade \
35+
jupyterlab \
36+
matplotlib \
37+
nltk \
38+
opencv-python-headless \
39+
pillow \
40+
pytorch-lightning \
41+
scikit-learn \
42+
scikit-image \
43+
seaborn \
44+
tensorboard \
45+
six
46+
47+
# Set up our notebook config.
48+
COPY jupyter_notebook_config.py /root/.jupyter/
49+
50+
# Jupyter has issues with being run directly:
51+
# https://github.com/ipython/ipython/issues/7062
52+
# We just add a little wrapper script.
53+
COPY run_jupyter.sh /
54+
55+
# IPython
56+
EXPOSE 8888
57+
58+
# Tensorboard
59+
EXPOSE 6006
60+
61+
WORKDIR /notebooks
62+
63+
CMD ["/run_jupyter.sh", "--allow-root"]

Diff for: ‎docker/pytorch1.11.0-cuda11.3.1/hooks/build

+5
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
#!/bin/bash
2+
# $IMAGE_NAME var is injected into the build so the tag is correct.
3+
docker build --build-arg VCS_REF=`git rev-parse --short HEAD` \
4+
--build-arg BUILD_DATE=`date -u +”%Y-%m-%dT%H:%M:%SZ”` \
5+
-t $IMAGE_NAME .

Diff for: ‎docker/pytorch1.11.0-cuda11.3.1/jupyter_notebook_config.py

+18
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
import os
2+
from IPython.lib import passwd
3+
4+
c = c # pylint:disable=undefined-variable
5+
c.NotebookApp.ip = '0.0.0.0'
6+
c.NotebookApp.port = int(os.getenv('PORT', 8888))
7+
c.NotebookApp.open_browser = False
8+
9+
# sets a password if PASSWORD is set in the environment
10+
if 'PASSWORD' in os.environ:
11+
password = os.environ['PASSWORD']
12+
if password:
13+
c.NotebookApp.password = passwd(password)
14+
else:
15+
c.NotebookApp.password = ''
16+
c.NotebookApp.token = ''
17+
del os.environ['PASSWORD']
18+

Diff for: ‎docker/pytorch1.11.0-cuda11.3.1/run_jupyter.sh

+2
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
#!/bin/bash
2+
jupyter notebook "$@"

Diff for: ‎docker/pytorch1.11.0/Dockerfile

+63
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,63 @@
1+
FROM wqael/docker:py3.8
2+
3+
LABEL description="JupyterLab / PyTorch 1.11.0 / Python 3.8.10 / ubuntu20.04" \
4+
maintainer="https://github.com/rlan/notebooks"
5+
6+
7+
# Build-time metadata as defined at http://label-schema.org
8+
ARG BUILD_DATE
9+
ARG VCS_REF
10+
ARG VERSION
11+
LABEL org.label-schema.build-date=$BUILD_DATE \
12+
org.label-schema.vcs-ref=$VCS_REF \
13+
org.label-schema.vcs-url="https://github.com/rlan/notebooks" \
14+
org.label-schema.version=$VERSION \
15+
org.label-schema.schema-version="1.0"
16+
17+
ARG PYTHON=python3
18+
ARG PIP=pip3
19+
# Skips tzdata configuration
20+
ARG DEBIAN_FRONTEND=noninteractive
21+
22+
RUN apt-get -qq update \
23+
&& apt-get -qq install -y --no-install-recommends \
24+
python3-pandas \
25+
&& apt-get -qq clean \
26+
&& rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*
27+
28+
RUN ${PIP} --no-cache-dir install -q --upgrade pip setuptools wheel
29+
30+
# Install pytorch
31+
RUN ${PIP} --no-cache-dir install -q --upgrade \
32+
torch==1.11.0+cpu torchvision==0.12.0+cpu torchaudio==0.11.0+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html
33+
34+
RUN ${PIP} --no-cache-dir install -q --upgrade \
35+
jupyterlab \
36+
matplotlib \
37+
nltk \
38+
opencv-python-headless \
39+
pillow \
40+
pytorch-lightning \
41+
scikit-learn \
42+
scikit-image \
43+
seaborn \
44+
tensorboard \
45+
six
46+
47+
# Set up our notebook config.
48+
COPY jupyter_notebook_config.py /root/.jupyter/
49+
50+
# Jupyter has issues with being run directly:
51+
# https://github.com/ipython/ipython/issues/7062
52+
# We just add a little wrapper script.
53+
COPY run_jupyter.sh /
54+
55+
# IPython
56+
EXPOSE 8888
57+
58+
# Tensorboard
59+
EXPOSE 6006
60+
61+
WORKDIR /notebooks
62+
63+
CMD ["/run_jupyter.sh", "--allow-root"]

Diff for: ‎docker/pytorch1.11.0/hooks/build

+5
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
#!/bin/bash
2+
# $IMAGE_NAME var is injected into the build so the tag is correct.
3+
docker build --build-arg VCS_REF=`git rev-parse --short HEAD` \
4+
--build-arg BUILD_DATE=`date -u +”%Y-%m-%dT%H:%M:%SZ”` \
5+
-t $IMAGE_NAME .

Diff for: ‎docker/pytorch1.11.0/jupyter_notebook_config.py

+18
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
import os
2+
from IPython.lib import passwd
3+
4+
c = c # pylint:disable=undefined-variable
5+
c.NotebookApp.ip = '0.0.0.0'
6+
c.NotebookApp.port = int(os.getenv('PORT', 8888))
7+
c.NotebookApp.open_browser = False
8+
9+
# sets a password if PASSWORD is set in the environment
10+
if 'PASSWORD' in os.environ:
11+
password = os.environ['PASSWORD']
12+
if password:
13+
c.NotebookApp.password = passwd(password)
14+
else:
15+
c.NotebookApp.password = ''
16+
c.NotebookApp.token = ''
17+
del os.environ['PASSWORD']
18+

Diff for: ‎docker/pytorch1.11.0/run_jupyter.sh

+2
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
#!/bin/bash
2+
jupyter notebook "$@"

0 commit comments

Comments
 (0)