Download from pytorch ci script (#2397)
Summary:
install_cuda.sh has been moved from pytorch/builder to pytorch/pytorch:
https://github.com/pytorch/builder/pull/1942
https://github.com/pytorch/pytorch/pull/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/{}'
```
Pull Request resolved: https://github.com/pytorch/benchmark/pull/2397
Test Plan: https://github.com/pytorch/benchmark/actions/runs/10163701096
Reviewed By: atalman
Differential Revision: D60454551
Pulled By: xuzhao9
fbshipit-source-id: 03322314b190e0fb95e7d7741152de65ab35ecd4
