pytorch
torchvision
cudatoolkit=10.1
transformers
accelerate
SentencePiece