When PyTorch supports CUDA 10.1?

From NVIDIA,It seems that the CUDA 10.1’s performance has a significant improvement over CUDA 10, When pytorch supports CUDA 10.1 ?



I am pretty sure you can compile from source for CUDA 10.1 with no problem. Let us know here if you have any issue with it.
Building binaries for all cuda versions is a bit complex as they are so many of them, that is why there is only a few of them supported as binaries.

1 Like

Got it! Thank you very much!

I actually have the same question and wanted to clear something out. The documentation mentions that we have to run the following command to install dependencies conda install -c pytorch magma-cuda92

Should we run this with magma-cuda10, since magma-cuda101 is not available?


I have verified that one cannot build PyTorch with CUDA 10.1, after a few rigorous attempts. The main reason is that MAGMA does not support 10.1 yet.


Will you please report back when you managed to build PyTorch with CUDA 10.1? Thanks in advance.

Does not #17874 show that it is possible to install PyTorch with CUDA 10.1?

It really does’t at all. Even if the building/installation succeeds, the actual usage will get you errors as well. I forgot exactly which error (I attempted this a couple of months back), but it’s unsolvable until official magma-cuda101 becomes available.


Thank you very much this is very helpful.

Thanks for telling us! Should really be set somewhere clear that CUDA 10.1 won’t work, tried for a long time to make it work :sweat_smile:

Does anyone know when CUDA 10.1 will be supported with pip install torch commands? Thanks.