no GPU error

[penguincute]

Hi!

After running:
python dreamer.py --logdir ~/logdir/atari_pong/dreamerv2/1 --configs defaults atari --task atari_pong

I got this error:

Traceback (most recent call last):
File "dreamer.py", line 324, in
main(parser.parse_args(remaining))
File "dreamer.py", line 239, in main
assert tf.config.experimental.list_physical_devices('GPU'), message
AssertionError: No GPU found. To actually train on CPU remove this assert.

Can you help me find the problem?

[Shawn5200]

so how to solove this problem？

[danijar]

The problem is unrelated to DreamerV2. Make sure you have a TF 2 and cuda setup correctly.