ubuntu系统下的大模型环境搭建

部署运行你感兴趣的模型镜像

miniconda安装

参考文章:Ubuntu系统上快速安装Miniconda的步骤_ubuntu 安装miniconda

下载最新版Miniconda:

wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh


给下载的文件添加可执行权限:
chmod +x Miniconda3-latest-Linux-x86_64.sh

运行安装文件:
./Miniconda3-latest-Linux-x86_64.sh 

验证是否安装成功:
conda --version

更新conda到最新版本:
conda update conda

虚拟环境搭建

创建虚拟环境

conda create -n env_name python=X.XX
conda create -n myPytorch python=3.12

虚拟环境创建成功之后,进行包的安装

  • torch安装,选择安装的是cuda12.8的版本

       进入torch官网:
       点击进入torch官网

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128

  • 安装transformers、modelscope等包
pip install transformers peft accelerate

pip install modelscope

       首先安装uv:pip install uv

       然后安装vllm:uv pip install vllm --torch-backend=auto

 官网参考:
You can install vLLM using either pip or uv pip:

# Install vLLM with CUDA 12.8.
# If you are using pip.
pip install vllm --extra-index-url https://download.pytorch.org/whl/cu128
# If you are using uv.
uv pip install vllm --torch-backend=auto

We recommend leveraging uv to automatically select the appropriate PyTorch index at runtime by inspecting the installed CUDA driver version via --torch-backend=auto (or UV_TORCH_BACKEND=auto). To select a specific backend (e.g., cu126), set --torch-backend=cu126 (or UV_TORCH_BACKEND=cu126). If this doesn't work, try running uv self update to update uv first.

  • 最终安装的核心包的版本如下:
torch                             2.7.1+cu128
torchaudio                        2.7.1+cu128
torchvision                       0.22.1+cu128
transformers                      4.54.0
vllm                              0.10.0
peft                              0.16.0
accelerate                        1.9.0
modelscope                        1.28.1

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch 2.5

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值