Hanjun-Dai / graph_comb_opt

Implementation of "Learning Combinatorial Optimization Algorithms over Graphs"
https://arxiv.org/abs/1704.01665
MIT License
491 stars 135 forks source link

when i run ./run_nstep_dqn.sh and minist ,the problem appear. #18

Open no-wings opened 4 years ago

no-wings commented 4 years ago

image image

Hanjun-Dai commented 4 years ago

It seems you don't have cuda enabled? the error says the malloc on gpu is failed.

no-wings commented 4 years ago

thanks,i have successfully run the code.