search

tsinghua-fib-lab / CLSR

The official implementation of "Disentangling Long and Short-Term Interests for Recommendation" (WWW '22)
MIT License
70 stars 19 forks source link

模型训练问题 #20

Closed xuhao-anhe closed 1 year ago

xuhao-anhe commented 1 year ago

作者您好，为什么我每次训练都是用的CPU跑的，更改了GPU-id，还是不行，请问怎么解决

DavyMorgan commented 1 year ago

请确保安装了GPU版本的TensorFlow，运行的时候可以通过CUDA_VISIBLE_DEVICES来指定GPU