Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Download pytorch ci script to install cuda #2397

Closed
wants to merge 1 commit into from
Closed

Download pytorch ci script to install cuda #2397

wants to merge 1 commit into from

Conversation

xuzhao9
Copy link
Contributor

@xuzhao9 xuzhao9 commented Jul 30, 2024 •
edited
Loading

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

@xuzhao9 xuzhao9 temporarily deployed to docker-s3-upload July 30, 2024 13:52 — with GitHub Actions Inactive
@xuzhao9 xuzhao9 temporarily deployed to docker-s3-upload July 30, 2024 13:52 — with GitHub Actions Inactive
@xuzhao9 xuzhao9 temporarily deployed to docker-s3-upload July 30, 2024 13:53 — with GitHub Actions Inactive
@xuzhao9 xuzhao9 requested review from atalman and kit1980 July 30, 2024 15:40
@facebook-github-bot
Copy link
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
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
Reviewers

@kit1980 kit1980 kit1980 approved these changes

@atalman atalman Awaiting requested review from atalman

Assignees
No one assigned
Labels
cla signed Merged
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
3 participants
@xuzhao9 @facebook-github-bot @kit1980