unsloth 部署(简单易上手版本)

装了一天终于装好了。。。。试了包括CSDN自己的平台,看了好多帖子,没想到最后用很简单的方法就装好了

  • 显卡: NVIDIA 显卡,显存建议至少 8GB (示例中使用 RTX 4090 24GB))。
  • Python 版本: 推荐 Python 3.11 (截至编写时,较新版本可能存在兼容性问题)。
  • CUDA: 需要预先安装与你NVIDIA驱动兼容的CUDA Toolkit。本教程以CUDA 11.8 (cu118) 为例。
  • audl算力平台

1. 创建 Conda 虚拟环境

首先,创建一个独立的 Conda 环境以隔离项目依赖。请确保文件夹路径不包含中文字符,以避免潜在问题。

打开你的命令行终端 (如 PowerShell 或 CMD),执行成功的话,命令行最前面应该有一个(unslothAmr)

bash 复制代码
conda create -n unslothAmr python=3.11
conda init bash
conda activate unslothAmr

2. 安装 Unsloth

bash 复制代码
pip install unsloth -i https://pypi.tuna.tsinghua.edu.cn/simple

3. 检查torch版本是CPU还是GPU

python 复制代码
import torch

# 检查 CUDA 是否可用
print("CUDA available:", torch.cuda.is_available())

# 如果可用,打印 CUDA 设备数量和当前设备名称
if torch.cuda.is_available():
    print("Number of CUDA devices:", torch.cuda.device_count())
    print("Current CUDA device:", torch.cuda.current_device())
    print("CUDA device name:", torch.cuda.get_device_name(torch.cuda.current_device()))
else:
    print("CUDA is not available. Running on CPU.")

如果显示

python 复制代码
CUDA available: True
Number of CUDA devices: 1
Current CUDA device: 0

继续下一步

bash 复制代码
pip install ipykernel
python -m ipykernel install --user --name=unsloth-kernel --display-name="Python (unslothAmr)"

输出结果

bash 复制代码
Installed kernelspec unsloth-kernel in /root/.local/share/jupyter/kernels/unsloth-kernel

后,新建一个notebook,选择内核 "Python (unslothAmr)",测试

相关推荐
信马堂4 小时前
WorkBuddy 接入第三方模型 API 全流程:安装与配置实录
人工智能·大模型·token·ai算力·workbuddy
朝朝辞暮i5 小时前
VLA 系统学习第 4 课:一个 Batch 进入神经网络后,模型到底是怎么“学会”的?
人工智能·python·神经网络·vla
IT古董5 小时前
《FDE前沿部署工程师实战教程》33 - Enterprise AI Observability:从Agent Trace到全链路智能运维
数据库·人工智能
xsd202411185 小时前
地网腐蚀AI识别算法全解析:从锈蚀图像到地下腐蚀快速定位
人工智能
oooost5 小时前
pytorch学习笔记2(transformer)
人工智能·机器学习
云杂项6 小时前
Exploring Model Inversion Attacks in the Black-box Setting(个人笔记)
人工智能
还卿一钵无情泪6 小时前
Unsloth 微调 构建自己的大模型 没有GPU也能微调
linux·开发语言·人工智能·python·大模型·nlp·unsloth
冬奇Lab7 小时前
LLM 驱动的自动化测试系列(06):移动端自动化(二)——DroidRun/Mobilerun 的角色级模型拆分
人工智能·测试
冬奇Lab7 小时前
一天一个开源项目(第230篇):HandRaw-Style —— 把 327 种手绘风格、165 种排版、36 种配色编号化,让 AI 画图不再每次都飘
人工智能·开源·资讯
罗西的思考7 小时前
从陶哲轩的访谈看:AI 数学研究方法论 & 给其它行业的启示
人工智能