Fix the CUDA error in train_eval_atari.py by creating the dataset and experience tensor on CPU, as suggested by Jiri.
Fix the CUDA error in train_eval_atari.py by creating the dataset and experience tensor on CPU, as suggested by Jiri.

PiperOrigin-RevId: 230397308
Change-Id: I964fc96edbc7427fd0af10a6f2eb790d28657105

tensorflow/agents,tensorflow/agents