Commit b78b5ea5 authored by Michael Carilli's avatar Michael Carilli
Browse files

Editing Dockerfile

parent 98fa5a3b
FROM pytorch/pytorch:0.4_cuda9_cudnn7
# Base image must at least have nvcc and CUDA installed.
FROM <base>
WORKDIR /workspace
# uninstall Apex if present
RUN pip uninstall -y apex || :
......
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