PyTorch with CUDA 11 compatibility - PyTorch Forums
discuss.pytorch.org › t › pytorch-with-cuda-11Jul 15, 2020 · Recently, I installed a ubuntu 20.04 on my system. Since it was a fresh install I decided to upgrade all the software to the latest version. So, Installed Nividia driver 450.51.05 version and CUDA 11.0 version. To my surprise, Pytorch for CUDA 11 has not yet been rolled out. My question is, should I downgrade the CUDA package to 10.2 or go with PyTorch built for CUDA 10.2 without downgrading ...