nzc/dnn_ctr

关于DeepFM

Opened this issue · 1 comments

ydx00 commented

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

nzc commented

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