Pytorch.org default installation command have wrong version of cuda

Site Feedback
1

Screenshot from 2021-03-08 09-10-12
Screenshot from 2021-03-08 09-10-12801×305 27 KB

pytorch.org default installation command have wrong version of cuda.

CUDA version 11.1 is chosen default but the command is 10.2.

2

Thanks for reporting it! It seems the highlighted CUDA version doesn’t match the default output.
If you reselect CUDA11.1, you’ll get the right install command.

CC @jspisak

3

Ok yeah this a weird one and something to do with .js refreshing. If you click back and forth on 10.2 and 11.1, the command updates and you get the correct one for 11.1. Our web team is looking into it. Thanks for raising this!

4

Here is the PR that fixes the issue. thanks for catching it!

1 Like