Hugging Face transformers 库依赖的 CUDA 是什么？如何安装？
galtjay · 2023-08-31 11:50:03 +08:00 · 530 次点击这是一个创建于 450 天前的主题，其中的信息可能已经有所发展或是发生改变。
最近由于工作的原因，需要去尝试部署海外大厂的开源 LLM 模型，进行一些测试验证。逐步呢就会接触到 AI 生态中的一些概念。
Hugging Face 的 transformers 库提供了使用 PyTorch 实现的预训练 NLP 模型，这些模型通过 CUDA 支持在 GPU 上进行训练和推理。
因此对于私有化部署而言，最核心的就是把 GPU 的驱动及 CUDA 都安装好，那么 CUDA 是什么？如何完成安装？
大家可以收藏一下，后期本地部署的时候，应该能帮大家节省时间。
Select Language