ARTICLE · INTELLIGENCE

战地情报 · 详情页

来自尧图项目组的一线实战观察与深度解析

Docker 部署Unsloth 绕开环境配置难题

Docker 部署Unsloth 绕开环境配置难题 前提GPU环境安装了NVIDIA Container Toolkit一般情况下apt-get update apt-get install -y nvidia-container-toolkit这个过程需要版本匹配等各种问题需要大家自己耐心安装开始首先建立一个环境隔离防止环境污染具体参考Unsloth 微调 构建自己的大模型 没有GPU也能微调-CSDN博客sudo apt update sudo apt install python3-venv source .venv/bin/activate 激活环境 使用完毕后记得关闭 deactivate下载Unsloth镜像docker pull unsloth/unsloth:latest这里面包括了所有 Unsloth 的示例 notebook和jupyter涵盖模型微调和强化学习如果只想使用基本功能只需要下载核心内容后面的标签变为docker pull unsloth/unsloth:core启动Unsloth启动GPU Unslothdocker run -it --gpus all \ -p 8080:8080 \ --name unsloth \ -v home/workspace:/root/workspace \ unsloth/unsloth:latest启动CPU Unsloth注意GPU参数改为-e UNSLOTH_ALLOW_CPU1docker run -it \ -p 8080:8080 \ -e UNSLOTH_ALLOW_CPU1 \ --name unsloth \ -v home/work:/root/workspace \ unsloth/unsloth:core如果在CPU的环境下使用GPU启动会出现以下提醒WARN: UNSLOTH_ALLOW_CPU1 and no GPU visible -- continuing on CPU. WARN: CPU mode covers Jupyter, GGUF tooling and llama.cpp (GGUF) Unsloth Studio chat. WARN: Training and loading Unsloth models (FastLanguageModel) still require an NVIDIA GPU.大概意思就是 只有GPU参与运算Jupyter, GGUF llama.cpp (GGUF) Unsloth Studio chat这几个模块运行没问题但是如果使用FastLanguageModel这个模块依然需要GPU配置好环境后可以进行微调模型可以参考Unsloth 微调 构建自己的大模型 没有GPU也能微调-CSDN博客
RELATED READING

延伸阅读

更多一线实战笔记与深度复盘,助您持续精进