Deep Learning with RTX 3090 (CUDA, cuDNN, Tensorflow, Keras, PyTorch)

Nvidia Driver

The RTX30-series has the Ampere architecture, therefore it will only work with Driver 450+ versions only.

Nvidia CUDA

Since we are now restricted by the driver version, we can only go for CUDA 11.0+.

Nvidia cuDnn

With CUDA 11.0+, we can only use cuDnn 8.0+.

Python

Spoiler alert: you will need to use tensorflow 2.5
Given the spoiler, you need to use Python3.8+

Tensorflow

There are currently 3 options to get tensorflow without with CUDA 11
参见官网

Pytorch

Pytorch actually released a new stable version 1.7.0 one day before I started writing this article, and it is now officially supporting CUDA 11
参见官网

---

参考

1. The Simple Guide: Deep Learning with RTX 3090 (CUDA, cuDNN, Tensorflow, Keras, PyTorch)
2. The Ultimate Guide: Ubuntu 18.04 GPU Deep Learning Installation (CUDA, cuDNN, Tensorflow, Keras, Opencv, PyTorch)