ollama运行怎么使用8张A10的GPU卡

为了使用 8 张 A10 GPU,你需要确保正确配置 CUDA_VISIBLE_DEVICES 环境变量来指定使用的 GPU。下面是一些步骤帮助你设置:

  1. 确认 GPU 环境
    使用 nvidia-smi 来确认你的 8 张 A10 GPU 是否被正确识别和安装驱动。

    nvidia-smi
    

    你应该看到每个 GPU 的详细信息,包括使用情况、显存、温度等。

  2. 设置环境变量
    如果你想指定使用特定的 GPU,可以通过设置 CUDA_VISIBLE_DEVICES 来选择。例如,如果你想让 Ollama 使用前 4 张 GPU(编号 0 到 3),可以在启动命令前设置该变量:

    export CUDA_VISIBLE_DEVICES=0,1,2,3
    

    如果你想使用所有 8 张 GPU,可以这么做:

    export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,
评论 15
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

MonkeyKing.sun

对你有帮助的话,可以打赏

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值