ServiceNow / N-BEATS

N-BEATS is a neural-network based model for univariate timeseries forecasting. N-BEATS is a ServiceNow Research project that was started at Element AI.

Other

508 stars 116 forks source link

Cannot run on gpu #11

Closed Dolores2333 closed 3 years ago

Dolores2333 commented 3 years ago

I can successfully make run the experiment on tourism dataset on cpu. However, when I use make run command=storage/experiments/tourism_interpretable/repeat=0,lookback=2,loss=MAPE/command gpu=0

I have the following feedback: the input device is not a TTY Makefile:30: recipe for target 'run' failed make: *** [run] Error 1

Dolores2333 commented 3 years ago

I found the following solution works with my problem:

change the part of pytorch installation in the Dockerfile

from:

Main frameworks

RUN pip install torch

to:

Main frameworks

RUN conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch-nightly