This repository has been archived by the owner on Nov 25, 2024. It is now read-only.
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
change pytorch cu121 to stable to fix ci (#97)
This PR is to fix wheel-test in CI. @chuangz0 also helped find the root course: Pytorch cu121 nightly seems will initialize CUDA at import time, which will cause later forked subprocess failed in CUDA calls. Fix: Change pytorch cu121 back to stable. Authors: - https://github.com/dongxuy04 Approvers: - Chuang Zhu (https://github.com/chuangz0) - Tingyu Wang (https://github.com/tingyu66) - Brad Rees (https://github.com/BradReesWork) - Ray Douglass (https://github.com/raydouglass) URL: #97
- Loading branch information