[FIXED] Training with TF 1.15 on RTX 3090

Issue

During I test several cases, I have some questions. One of that is "Training with tf 1.15 on RTX 3090".

[MY current environments]

1. python : v3.7.9
2. tensorflow : v1.15.5
3. cuda : v11.2
4. cudnn : v8.1.0
5. os : window 10

Can I proceed with the training in this environment?

Solution

I believe this cannot work. Tensorflow 1.15 requires cuda 10:

https://www.tensorflow.org/install/source#gpu

I believe the RTX 3090 does not support cuda 10.

You might need to use either TF 1.15 on CPU. Or a 2.x TF release with a more recent cuda version.

Answered By - Filippo Grazioli