Download pytorch ci script to install cuda

[xuzhao9]

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]

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

[facebook-github-bot]

@xuzhao9 merged this pull request in 67b5e75.