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

RuntimeError: cuda runtime error (2) : out of memory #2

Closed me2george closed 6 years ago

me2george commented 6 years ago

while running main.py, I got some error described in title. Any solutions or suggestions for this issue? My enviroment: windows10 with Intel(R) Core(TM) i5-6400 CPU @ 2.70GHz 16GB RAM cuda9.0 installed(with cudnn) with NVIDIA GTX 750TI. And I use the code for pytorch 0.4.

Many thanks in advance.

image

cjbarb7 commented 6 years ago

From the Readme file: "Running out-of-the-box with our pre-trained models using GPU acceleration requires 8GB of GPU memory"

Your 750 Ti doesnt have enough memory.