1. 将环境迁移到win,可以直接下载修改好的cuda环境https://download.csdn.net/download/liangjiubujiu/89368302?spm=1001.2014.3001.5503
2. 修改hugging face的缓存路径
3. 重启pycharm
4. 编写测试代码,注意可能由于网络链接问题,需要关闭代理并使用镜像网站解决。
import os
os.environ["HF_ENDPOINT"] = "https://hf-mirror.com"
5.编写代码测试
import os
os.environ["HF_ENDPOINT"] = "https://hf-mirror.com"
from transformers import pipeline
pipe = pipeline("text-generation", model="bigscience/bloom-560m")
print(pipe(
"电子科技大学是我国一所专",
max_length=30,
))
6. 设置hf环境下的python.exe为编译器并运行结果
[{'generated_text': '电子科技大学是我国一所专职的综合性大学,是教育部直属的全国重点大学,也是国家“211工程”重点建设高校,是'}]