How can I run it on a gpu?

gsurma / atari

AI research environment for the Atari 2600 games 🤖.

https://gsurma.github.io

MIT License

253 stars 81 forks source link

Open THSWind opened 4 years ago

THSWind commented 4 years ago

I have write os.environ["CUDA_VISIBLE_DEVICES"] = "0" in atari.py, but it doesn't work, help me plz