Hi,
Here is the info:
RTX2080-Ti (Driver: 442.19)
Tensorflow 2.1.0
Python 3.6.8
Cuda 10.1
Even if I tried to reduce the batch size, the screen still blacks out either at the beginning or in the middle of training.
Any suggestions are appreciated!