-
Notifications
You must be signed in to change notification settings - Fork 374
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Creating pull request of hacks I needed to run sample.py on cuda #178
Conversation
Hi @DavidLKing did you try the |
Hi @pskrunner14! I did, but it still fails when I try to run
I'm fine with this not being a full merge. Would an issue make more sense? |
Hey @DavidLKing are you sure you have the latest changes? It seems to be working fine on my side:
|
Okay, it seems there have been more drastic changes. I was using an older version of |
Hi @DavidLKing yes there have been quite a few changes in the core API. Hopefully by next month it will be ready, still have a few bugs in the new features. |
When I run even the basic
sample.py
script underexamples
on a cuda enabled machine, I still get errors that not all the vectors are cuda vectors (some are still cpu). You can ignore my edits insample.py
, but I did annotate in each place where I needed to addvector = vector.cuda()
to makesample.py
run. This occurs even whentorch.device('cuda')
is called, which should not be the case in PyTorch 0.4.0+.Thank you, and feel free to follow up with any questions.