Pytorh 1.9 with cuda 11.0

1

Can i install pytorch 1.9 or 1.10 with cuda 11.0? Will it work or give me errors?
Currently i have cuda 11.0 install with tensorflow 2.4 and pytorch 1.7.1.
I could not find any other cuda version which is compatible for both TF and Pytorch.

So if i install pytorch 1.9? will it cause any issue?

I am using ubuntu and dont want to install conda

2

Yes, you can build PyTorch 1.9 and 1.10 with CUDA 11.0.
The current binaries ship with CUDA 10.2, 11.3, and 11.5, so you would need to build from source if you need this particular CUDA version.

PyTorch should run, but I haven’t checked the compatibility with other frameworks and same/different CUDA versions.