Skip to content

GitLab

Commit acf25fc8 authored by DeepMind's avatar DeepMind Committed by Copybara-Service
Browse files

Switch to using cudnn base image. 

PiperOrigin-RevId: 387058301
Change-Id: If7d2716bfe89228e9fb8f909b72c425bd89f1f7f
parent 2028cd0a
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
docker/Dockerfile
View file @ acf25fc8
...@@ -13,7 +13,7 @@ ...@@ -13,7 +13,7 @@
# limitations under the License. # limitations under the License.
ARG CUDA=11.0 ARG CUDA=11.0
FROM nvidia/cuda:${CUDA}-base FROM nvidia/cuda:${CUDA}-cudnn8-runtime-ubuntu16.04
# FROM directive resets ARGS, so we specify again (the value is retained if # FROM directive resets ARGS, so we specify again (the value is retained if
# previously set). # previously set).
ARG CUDA ARG CUDA
...@@ -52,7 +52,7 @@ ENV PATH="/opt/conda/bin:$PATH" ...@@ -52,7 +52,7 @@ ENV PATH="/opt/conda/bin:$PATH"
RUN conda update -qy conda \ RUN conda update -qy conda \
&& conda install -y -c conda-forge \ && conda install -y -c conda-forge \
openmm=7.5.1 \ openmm=7.5.1 \
cudatoolkit==${CUDA}.3 \ cudatoolkit==${CUDA_VERSION} \
pdbfixer \ pdbfixer \
pip \ pip \
python=3.7 python=3.7
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment