python==3.9.15 pytorch==1.10.2 pytorch-lightning==0.9.0 transformers==4.15.0
- start with a clean python3 virtual environment with conda
- install pytorch with
conda install pytorch==1.10.2 pytorch-cuda=11.6 -c pytorch -c nvidia - install rest of the packages:
pip install pytorch-lightning==0.9.0 transformers==4.15.0 tokenizers