"配置了GPU卡是3,4,5;但实际用的是0,1,2;会导致ModelScope进程就不能运行了,怎么处理?
"
" 这行代码放到最前面
import torch之前。
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '1,2,3'
放脚本第一二行。此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”"
ModelScope旨在打造下一代开源的模型即服务共享平台,为泛AI开发者提供灵活、易用、低成本的一站式模型服务产品,让模型应用更简单!欢迎加入技术交流群:微信公众号:魔搭ModelScope社区,钉钉群号:44837352