"vscode:/vscode.git/clone" did not exist on "bdd690e555efc60092ac0c255b69bca1793a047f"
Dockerfile 387 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
FROM pytorch/pytorch:0.4_cuda9_cudnn7
WORKDIR /workspace
# uninstall Apex if present
RUN pip uninstall -y apex || :
# SHA is something the user can alter to force recreation of this Docker layer, 
# and therefore force cloning the latest version of Apex
RUN SHA=43d1ae08 git clone https://github.com/NVIDIA/apex.git
WORKDIR /workspace/apex
RUN python setup.py install
WORKDIR /workspace