开发者社区 > ModelScope模型即服务 > 正文

ModelScope中,修改了代码还是报错怎么解决?

"单卡内存不足,多卡的话,报错RuntimeError: Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing, you must use the 'spawn' start method。ModelScope中,修改了代码还是报错怎么解决? import os
import subprocess
import multiprocessing as mp

if name == 'main':

# 设置多处理的启动方法为'spawn'
mp.set_start_method('spawn')

# 设置环境变量
os.environ['CUDA_VISIBLE_DEVICES'] = '2,3,4,5'
os.environ['RAY_memory_monitor_refresh_ms'] = '0'

# 运行swift deploy命令
subprocess.run([
    'swift', 'app-ui',
    '--model_type', 'qwen2-7b',
    '--infer_backend', 'vllm',
    '--tensor_parallel_size', '4',
    '--dtype', 'fp16'
]) 还是报错RuntimeError: Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing, you must use the 'spawn' start method"

展开
收起
小小爱吃香菜 2024-07-17 08:15:35 96 0
0 条回答
写回答
取消 提交回答

ModelScope旨在打造下一代开源的模型即服务共享平台,为泛AI开发者提供灵活、易用、低成本的一站式模型服务产品,让模型应用更简单!欢迎加入技术交流群:微信公众号:魔搭ModelScope社区,钉钉群号:44837352

相关电子书

更多
视觉AI能力的开放现状及ModelScope实战 立即下载
ModelScope助力语音AI模型创新与应用 立即下载
低代码开发师(初级)实战教程 立即下载