Skip to content

GitLab

Unverified Commit 4ef4fabf authored by mcarilli's avatar mcarilli Committed by GitHub
Browse files

Update Dockerfile

parent 7d05704c
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/docker/Dockerfile
View file @ 4ef4fabf
# Base image must at least have pytorch and CUDA installed.
ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:18.04-py3
ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:18.12-py3
FROM $BASE_IMAGE
ARG BASE_IMAGE
RUN echo "Installing Apex on top of ${BASE_IMAGE}"
......
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