search

nnzhan / MTGNN

MIT License
798 stars 222 forks source link

Struggling to run single step caused by a cuda error #28

Open egeSPACEoztas opened 2 years ago

egeSPACEoztas commented 2 years ago

I am struggling to run the single step training code for exchange rate data. Getting an CUDA error.

asdfggg Here is the error exception in I get when I attempt to replicate the process detailed in the readme file.

I am using a windows 10 notebook for this. I believe I have installed all of the dependencies listed inside the repository, but I might be wrong.

gokayfem commented 2 years ago

change cuda:1 to cuda:0 in train_single_step.py

parser.add_argument('--device',type=str,default='cuda:1',help='')

pazars commented 1 year ago

Getting a similar error when training from Google Colab T4 environment. Running with 'cuda:0' works, but expected that this would run smoothly out of the box.