search

nzc / dnn_ctr

The framework to deal with ctr problem。The project contains FNN,PNN,DEEPFM, NFM etc
756 stars 285 forks source link

关于DeepFM #3

Open ydx00 opened 6 years ago

ydx00 commented 6 years ago

您好，我运行你的代码，报如下错误，请问怎么解决，第一次用pytorch RuntimeError: cuda runtime error (10) : invalid device ordinal at torch/csrc/cuda/Module.cpp:88

nzc commented 6 years ago

@ydx00 大概你配的是非gpu版的pytorch 你有两个选择：1.将代码的use_cuda设置成False 2.配置一个gpu版的pytorch