Skip to content

Commit

Permalink
PyTorch Dockerfile should update submodules recursively.
Browse files Browse the repository at this point in the history
Summary: Pull Request resolved: #21216

Differential Revision: D15584114

Pulled By: bddppq

fbshipit-source-id: dbe0c3a54024a90fcd2c6689f8b9689ed0cd639b
  • Loading branch information
xuhdev authored and facebook-github-bot committed May 31, 2019
1 parent 0aeb971 commit 4dbeb87
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docker/pytorch/Dockerfile
Expand Up @@ -23,7 +23,7 @@ ENV PATH /opt/conda/bin:$PATH
WORKDIR /opt/pytorch WORKDIR /opt/pytorch
COPY . . COPY . .


RUN git submodule update --init RUN git submodule update --init --recursive
RUN TORCH_CUDA_ARCH_LIST="3.5 5.2 6.0 6.1 7.0+PTX" TORCH_NVCC_FLAGS="-Xfatbin -compress-all" \ RUN TORCH_CUDA_ARCH_LIST="3.5 5.2 6.0 6.1 7.0+PTX" TORCH_NVCC_FLAGS="-Xfatbin -compress-all" \
CMAKE_PREFIX_PATH="$(dirname $(which conda))/../" \ CMAKE_PREFIX_PATH="$(dirname $(which conda))/../" \
pip install -v . pip install -v .
Expand Down

0 comments on commit 4dbeb87

Please sign in to comment.