allenai / ai2thor

An open-source platform for Visual AI.

Home Page:http://ai2thor.allenai.org

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to run ai2thor on NVIDIA GPU?

sun123-cmd opened this issue · comments

I tried to run codes as README, but it run on the CPU device. Can I try to move it on GPU to run faster?