Skip to content
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

pip version problems - updated requirements.txt #108

Open
jnehring opened this issue Jun 6, 2021 · 0 comments
Open

pip version problems - updated requirements.txt #108

jnehring opened this issue Jun 6, 2021 · 0 comments

Comments

@jnehring
Copy link

jnehring commented Jun 6, 2021

I ran into some strange problems and want to share my solution here. When I ran the training script it did not use the GPUs. Another problem was that distributed training froze during startup. I could solve the problems by adding version numbers for torch and pytorch-ignite to requirements.txt. So if you run into similar problems try using explicit version numbers when setting up your python environment.

Here is the new content of my requirements.txt:

torch==1.8.1
pytorch-ignite==0.3.0
transformers==2.5.1
tensorboardX==1.8
tensorflow  # for tensorboardX
spacy
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant