Skip to content

Commit

Permalink
revert to cuda9.2 to fix docker building problems
Browse files Browse the repository at this point in the history
  • Loading branch information
decrispell committed Jan 23, 2019
1 parent 8beddd9 commit 921a78e
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions docker/Dockerfile
Original file line number Diff line number Diff line change
@@ -1,10 +1,10 @@
FROM nvidia/cudagl:10.0-devel-ubuntu18.04
FROM nvidia/cudagl:9.2-devel-ubuntu18.04
ARG cpu_image=0
ENV CUDNN_VERSION 7.4.1.5
SHELL ["/usr/bin/env", "bash", "-euxvc"]
RUN apt-get update && apt-get install -y --no-install-recommends\
libcudnn7=$CUDNN_VERSION-1+cuda10.0 \
libcudnn7-dev=$CUDNN_VERSION-1+cuda10.0 && \
libcudnn7=$CUDNN_VERSION-1+cuda9.2 \
libcudnn7-dev=$CUDNN_VERSION-1+cuda9.2 && \
apt-mark hold libcudnn7
RUN apt-get update && apt-get install -y --no-install-recommends \
wget git unzip python3-dev cmake ninja-build libz-dev libssl-dev libboost-dev libglew-dev libtiff-dev python3-venv libjpeg-turbo8-dev \
Expand All @@ -21,7 +21,7 @@ RUN python3 -m venv /pix2face/venv;\
pip install --upgrade pip; \
pip install -r /pix2face/requirements.txt; \
if [ $CPU_ONLY -eq 0 ]; then \
pip install https://download.pytorch.org/whl/cu100/torch-1.0.0-cp36-cp36m-linux_x86_64.whl; \
pip install torch; \
else \
pip install https://download.pytorch.org/whl/cpu/torch-1.0.0-cp36-cp36m-linux_x86_64.whl; \
fi; \
Expand Down

0 comments on commit 921a78e

Please sign in to comment.