no GPU error
penguincute opened this issue · comments
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?
so how to solove this problem?
The problem is unrelated to DreamerV2. Make sure you have a TF 2 and cuda setup correctly.