Set sb3_imitation.py to CPU to fix error with CUDA

edbeeching / godot_rl_agents

An Open Source package that allows video game creators, AI researchers and hobbyists the opportunity to learn complex behaviors for their Non Player Characters or agents

MIT License

902 stars 63 forks source link

Set sb3_imitation.py to CPU to fix error with CUDA #196

Closed Ivan-267 closed 1 month ago

Ivan-267 commented 1 month ago

Set device to CPU as a temporary fix for CUDA users.

Thanks to @MichaelrMentele for reporting this issue and testing the solution.

Relevant PR to track (thanks to @JayRothenberger for finding it): https://github.com/HumanCompatibleAI/imitation/pull/831

Ivan-267 commented 1 month ago

@edbeeching

In short: Running the script with CUDA setup returned an error, and this PR should make the script functional for everyone (on CPU) until the issue is resolved. There's some data in the linked PR for tracking the issue.