训练模型第一步--在本地调用语言模型
最近几天一直在尝试在本地跑通语言模型--firefly,也是卡了很久,踩了很多坑。
1.安装anaconda,他是一个环境管理工具,我利用它创建多个虚拟环境,再激活环境,在其中安装环境和库
2.安装cuda(用与GPU训练模型),安装cunna(用于加速训练语言模型),安装pytorch(它是一个模型深度学习的框架,包含很多库和工具包)
安装transformers库等
3.配置vscode的编译器路径(虚拟环境的python),配置pytorch环境变量
4.下载fierefly模型,再vscode下运行代码调用