I have CUDA version 11.6 hence am using torch-1.12.0+cu116
But functorch requires torch<1.12, >=1.11
I think the functorch
release is still blocked by #80489 so you might need to build it from source for now.
I think the functorch
release is still blocked by #80489 so you might need to build it from source for now.