Closed Darkbblue closed 1 year ago
I've found a viable setup:
# create a new env with python=3.8
conda install pytorch==1.12.1 torchvision==0.13.1 cudatoolkit=10.2 -c pytorch
conda install nltk
python3 -m nltk.downloader punkt
conda install -c huggingface transformers
transformers has been updated and now it requires torch>=1.5.0