Skip to content

Commit

Permalink
Revert moving the cupti headers and library files (#2419)
Browse files Browse the repository at this point in the history
Summary:
It turns out that the failed Kineto workflow is not related to Torchbench docker. We do not need to manually move the cupti header files, so let's keep it simple.

Pull Request resolved: #2419

Reviewed By: aaronenyeshi

Differential Revision: D61284093

Pulled By: xuzhao9

fbshipit-source-id: 8a4f59888e1d7586c11743a15fc05ee5d9ab8a4f
  • Loading branch information
xuzhao9 authored and facebook-github-bot committed Aug 14, 2024
1 parent a2ea728 commit 23fe825
Showing 1 changed file with 0 additions and 7 deletions.
7 changes: 0 additions & 7 deletions docker/gcp-a100-runner-dind.dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -35,13 +35,6 @@ RUN sudo mkdir -p /workspace; sudo chown runner:runner /workspace
RUN cd /workspace; mkdir -p pytorch-ci; cd pytorch-ci; wget https://raw.githubusercontent.com/pytorch/pytorch/main/.ci/docker/common/install_cuda.sh
RUN sudo bash -c "set -x;export OVERRIDE_GENCODE=\"${OVERRIDE_GENCODE}\" OVERRIDE_GENCODE_CUDNN=\"${OVERRIDE_GENCODE_CUDNN}\"; bash /workspace/pytorch-ci/install_cuda.sh 12.4"

# Move CUPTI libraries to the cuda directory
RUN sudo bash -c "set -x; mv /usr/local/cuda/extras/CUPTI/include/* /usr/local/cuda/include"
RUN sudo bash -c "set -x; mv /usr/local/cuda/extras/CUPTI/lib64/* /usr/local/cuda/lib64"
# Move Debugger libraries to the cuda directory
RUN sudo bash -c "set -x; mv /usr/local/cuda/extras/Debugger/include/* /usr/local/cuda/include"
RUN sudo bash -c "set -x; mv /usr/local/cuda/extras/Debugger/lib64/* /usr/local/cuda/lib64"

# Install miniconda
RUN wget -q https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O /workspace/Miniconda3-latest-Linux-x86_64.sh
RUN cd /workspace && \
Expand Down

0 comments on commit 23fe825

Please sign in to comment.