search

danijar / dreamerv2

Mastering Atari with Discrete World Models
https://danijar.com/dreamerv2
MIT License
886 stars 195 forks source link

no GPU error #6

Closed penguincute closed 3 years ago

penguincute commented 3 years ago

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 commented 2 years ago

so how to solove this problem？

danijar commented 2 years ago

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