CUDA VERSION 11.1 大模型微调可用的torch及其他包
提供一个可行的cuda11.1 版本下可行的版本方案
CUDA11.1 可以用的对应的python 版本为3.8.19(3.8), torch版本为1.10.2+cu111, torchvision 版本为0.10.0+cu111, torchaudio 版本为0.10.2,accelerate 版本为0.31.0, transformers 版本为4.41.2
- transformers 如果版本版本过低会出现 Qwen2Tokenizer importError
- python3.6 最高能安装的transformers版本到4.1.8.0 不支持Qwen2的大模型微调