Skip to content

Instantly share code, notes, and snippets.

@ben0it8

ben0it8/Dockerfile

Created October 31, 2019 12:20
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save ben0it8/28ccae6296c63902789883a8814c8616 to your computer and use it in GitHub Desktop.
Save ben0it8/28ccae6296c63902789883a8814c8616 to your computer and use it in GitHub Desktop.
Download ZIP
Ubuntu 18.04 + Python3.7 + tf1.5 + dali-nightly
Raw
Dockerfile
FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu18.04
WORKDIR /srv
RUN chmod 777 /tmp
RUN apt-get update && apt-get install -y --no-install-recommends \
build-essential \
python3.7 python3.7-dev python3-virtualenv
ENV VENV=/srv/venv
RUN python3 -m virtualenv --python=/usr/bin/python3.7 $VENV
ENV PATH="${VENV}/bin:${PATH}"
# Install TF
RUN pip install --pre --no-cache-dir "tensorflow==1.15.*"
# Install DALI + TF plugin
RUN pip install --no-cache-dir --extra-index-url https://developer.download.nvidia.com/compute/redist/nightly/cuda/10.0 nvidia-dali-nightly
RUN pip install --no-cache-dir --extra-index-url https://developer.download.nvidia.com/compute/redist/nightly/cuda/10.0 nvidia-dali-tf-plugin-nightly
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment