windows11下的大模型部署 —— 环境准备(组件间版本对应)

1. cuda

更新显卡驱动(windows下安装即可,和wsl共享):官方驱动 | NVIDIA

显卡驱动版本和cuda版本对应:CUDA 12.4 Update 1 Release Notes (nvidia.com)

下载cuda toolkit(注意Distribution需要选择wsl):CUDA Toolkit 12.4 Update 1 Downloads | NVIDIA Developer

2. cuDNN

cuDNN Archive | NVIDIA Developer

根据cuda版本,选择对应的cuDNN进行解压,解压缩后替换cuda目录同名文件夹

3. torch

pytorch/vision: Datasets, Transforms and Models specific to Computer Vision (github.com)

Previous PyTorch Versions | PyTorch

根据python版本,选择torch版本,再根据cuda版本选择合适的pytorch-cuda

4. vllm

安装 — vLLM

因为vllm引入了其他cuda和torch版本,所以如果想使用vllm框架,则建议新建一个conda环境,再安装vllm

tips:清华镜像源 

pip install xxxx -i Simple Indexicon-default.png?t=N7T8https://pypi.tuna.tsinghua.edu.cn/simple

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值