Короче у меня не совсем с графикой была проблема. А с машинным обучением. Обучалка запускалась только на проце, а не видеокарте.

Проблема оказалась в несовпадении библиотек:
Cuda драйвер был версии 11, а тензорфлоу библиотека на тот момент ее не поддерживала 

Пришлось вручную снижать версию куда до совместимой (перед этим смотрел в интернете таблицы совместимости версий).

Возможно и здесь такая же проблема.
 Поищи в интернете инфу о совместимости всех драйверов и библиотек, которые скачиваются