Drop cuda 10.2 support for functorch 0.2.0 release

pytorch / functorch

functorch is JAX-like composable function transforms for PyTorch.

https://pytorch.org/functorch/

BSD 3-Clause "New" or "Revised" License

1.38k stars 102 forks source link

Closed zou3519 closed 2 years ago

zou3519 commented 2 years ago

This PR:

adds an error message when functorch is imported alongside a PyTorch CUDA 10.2 binary (pip or conda)
changes the CI to use cuda 11.3 for linux (since the error message triggers no matter if it's a source build or binary build)
Updates the links to PyTorch binaries to the officially released ones instead of using the pytorch-test channel.

Test Plan: