transfer-learning-conv-ai icon indicating copy to clipboard operation
transfer-learning-conv-ai copied to clipboard

pip version problems - updated requirements.txt

Open jnehring opened this issue 4 years ago • 0 comments

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

jnehring avatar Jun 06 '21 18:06 jnehring