search

andyzeng / visual-pushing-grasping

Train robotic agents to learn to plan pushing and grasping actions for manipulation with deep reinforcement learning.
http://vpg.cs.princeton.edu/
BSD 2-Clause "Simplified" License
898 stars 315 forks source link

KeyError: 'module name can\'t contain "."' #8

Open weiwuhuhu opened 5 years ago

weiwuhuhu commented 5 years ago

Hello, I run the simulation with GPU according to the tutorial. I had met the warning.The warning as follows: PyTorch no longer supports this GPU because it is too old. My GPU is 840m, The capability of compute is 5.0.Then I uninstall the pytorch. And install the pytorth by source. Then when I run the simulation,The error occured :KeyError: 'module name can't contain "."How can i run the simulation with GPU

andyzeng commented 5 years ago

The error :KeyError: 'module name can't contain "." comes from running the code in the master branch (compatible with PyTorch v0.3) with a later version of PyTorch (v0.4+). Try running the code from this branch in CPU mode (warning: very slow). Your GPU does not have enough memory to run the code in GPU mode, which requires 8GB+.

weiwuhuhu commented 5 years ago

Thank you very much,when I run the code in cpu, it often crashed. My computer is CPU:I5 and GPU: 840m, Now i want to buy a computer which is cpu: I5-8300h, GPU: 1050ti 4G. Because I think your method for grasp is so good. And does this have the method to run in 1050Ti, because the cpu is too slowly.