PyTorch with CUDA 11 compatibility

As explained here, the binaries are not built yet with CUDA11. However, the initial CUDA11 enablement PRs are already merged, so that you could install from source using CUDA11.

If you want to use the binaries, you would have to stick to 10.2 for now.

1 Like