Skip to content

Download pytorch ci script to install cuda#2397

Closed
xuzhao9 wants to merge 1 commit intomainfrom
xz9/fix-nightly-docker-build
Closed

Download pytorch ci script to install cuda#2397
xuzhao9 wants to merge 1 commit intomainfrom
xz9/fix-nightly-docker-build

Conversation

@xuzhao9
Copy link
Copy Markdown
Contributor

@xuzhao9 xuzhao9 commented Jul 30, 2024

install_cuda.sh has been moved from pytorch/builder to pytorch/pytorch:

pytorch/builder#1942
pytorch/pytorch#129022

We need to make sure that sm_90a is included:

xargs -I '{}' bash -c 'echo {} && /usr/local/cuda-12.4/bin/nvprune -gencode arch=compute_80,code=sm_80 -gencode arch=compute_86,code=sm_86 -gencode arch=compute_90,code=sm_90 -gencode arch=compute_90a,code=sm_90a /usr/local/cuda-12.4/lib64/{} -o /usr/local/cuda-12.4/lib64/{}'

Test plan: https://github.com/pytorch/benchmark/actions/runs/10163701096

@facebook-github-bot
Copy link
Copy Markdown
Contributor

@xuzhao9 has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

@xuzhao9 xuzhao9 changed the title Download from pytorch ci script Download pytorch ci script to install cuda Jul 30, 2024
@facebook-github-bot
Copy link
Copy Markdown
Contributor

@xuzhao9 merged this pull request in 67b5e75.

@xuzhao9 xuzhao9 deleted the xz9/fix-nightly-docker-build branch August 1, 2024 02:24
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants