Install torch Cuda 11.2 Ubuntu 20.04 GeForce RTX 3090

Hi everybody,

I have problems to use my GPUs with torch in Ubuntu 20.04 and my GeForce RTX 3090 and CUDA 11.2.

Does anybody have a tip or an idea how I can use my GPU?

Thanks and best regards,

1 Like

That helped me a lot: Install CUDA 11.2, cuDNN 8.1.0, PyTorch v1.8.0 (or v1.9.0), and python 3.9 on RTX3090 for deep learning | by Yifan Guo | Analytics Vidhya | Mar, 2021 | Medium

1 Like

Unfortunately, the article is behind a paywall…