Created
October 31, 2019 12:20
-
-
Save ben0it8/28ccae6296c63902789883a8814c8616 to your computer and use it in GitHub Desktop.
Ubuntu 18.04 + Python3.7 + tf1.5 + dali-nightly
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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