joeyNMT 1.1/Google Colab - uses device:cpu instead of GPU
See original GitHub issueProblem
When training using Masakhane custom data notebook, and pip installing joeyNMT,
!cd joeynmt; python3 -m joeynmt train configs/transformer_$src$tgt.yaml
uses CPU instead of GPU.
This occurs if I use version 1.1, but NOT if I pip install 1.0 specifically.
- If I use
joeynmt==1.0
when pip installing, I getdevice:cuda
- If I use
joeynmt
when pip installing, it installs 1.1, and I getdevice:cpu
System (please complete the following information):
- Google Colab, https://github.com/masakhane-io/masakhane-mt/blob/master/starter_notebook-custom-data.ipynb
- joeyNMT pip installed
Issue Analytics
- State:
- Created 3 years ago
- Reactions:1
- Comments:13 (3 by maintainers)
Top Results From Across the Web
joeynmt - bytemeta
joeynmt repo issues. ... Joey NMT for paper Correct Me If You Can ... joeyNMT 1.1/Google Colab - uses device:cpu instead of GPU....
Read more >Top Related Medium Post
No results found
Top Related StackOverflow Question
No results found
Troubleshoot Live Code
Lightrun enables developers to add logs, metrics and snapshots to live code - no restarts or redeploys required.
Start FreeTop Related Reddit Thread
No results found
Top Related Hackernoon Post
No results found
Top Related Tweet
No results found
Top Related Dev.to Post
No results found
Top Related Hashnode Post
No results found
Top GitHub Comments
There is a deprecation warning about a specific class in torchtext:
Is this perhaps a problem when updating the requirements to torchtext==0.8.1 ?
same issue! It has something to do with the torch version I believe but not sure yet