hidasib / GRU4Rec

GRU4Rec is the original Theano implementation of the algorithm in "Session-based Recommendations with Recurrent Neural Networks" paper, published at ICLR 2016 and its follow-up "Recurrent Neural Networks with Top-k Gains for Session-based Recommendations". The code is optimized for execution on the GPU.
Other
754 stars 223 forks source link

cuda error #44

Open outside-BUPT opened 3 years ago

outside-BUPT commented 3 years ago

I run this command:

$python run.py ../GRU4Rec_TensorFlow_beauty/data/train.tsv -t ../GRU4Rec_TensorFlow_beauty/data/valid.tsv -m 1 5 10 20 \ -ps loss=bpr-max,final_act=elu-0.5,hidden_act=tanh,layers=100,adapt=adagrad,n_epochs=10,batch_size=32,dropout_p_embed=0.0,dropout_p_hidden=0.0

But I get this error:

WARNING (theano.tensor.blas): Using NumPy C-API based implementation for BLAS functions. Can not use cuDNN on context None: cannot compile with cuDNN. We got this error: b'/tmp/try_flags_b5qyyrtx.c:4:10: fatal error: cudnn.h: No such file or directory\n #include \n ^~~~~\ncompilation terminated.\n'

Traceback (most recent call last): File "run.py", line 122, in gru.fit(data, sample_store=args.sample_store_size, store_type='gpu', ckpt_path=args.save_model) File "/hdd/lhz/cll/baselines/GRU4Rec+_beauty/gru4rec.py", line 553, in fit updates_st[self.ST] = gpu_searchsorted(P, X, dtype_int64=True).reshape((generate_length, self.n_sample)) File "/hdd/lhz/cll/baselines/GRU4Rec+_beauty/gpu_ops.py", line 43, in gpu_searchsorted return cto.GpuBinarySearchSorted(dtype_int64=dtype_int64)(P, X) File "/home/anaconda3/envs/theano/lib/python3.6/site-packages/theano/gof/op.py", line 615, in call node = self.make_node(*inputs, *kwargs) File "/hdd/lhz/cll/baselines/GRU4Rec+_beauty/custom_theano_ops.py", line 292, in make_node d = as_gpuarray_variable(d, context_name=self.context_name) File "/home/anaconda3/envs/theano/lib/python3.6/site-packages/theano/gpuarray/basic_ops.py", line 79, in as_gpuarray_variable return copy_stack_trace(x, GpuFromHost(context_name)(x)) File "/home/anaconda3/envs/theano/lib/python3.6/site-packages/theano/gof/op.py", line 615, in call node = self.make_node(inputs, **kwargs) File "/home/anaconda3/envs/theano/lib/python3.6/site-packages/theano/gpuarray/basic_ops.py", line 674, in make_node dtype=x.dtype)() File "/home/anaconda3/envs/theano/lib/python3.6/site-packages/theano/gpuarray/type.py", line 186, in init get_context(self.context_name) File "/home/anaconda3/envs/theano/lib/python3.6/site-packages/theano/gpuarray/type.py", line 104, in get_context raise ContextNotDefined("context name %s not defined" % (name,)) theano.gpuarray.type.ContextNotDefined: context name None not defined

CUDA Version 10.1.105

Anyone could help?