开发者社区 > ModelScope模型即服务 > 计算机视觉 > 正文

ModelScope这个镜像coda怎么没有运行起来呢?

ModelScope这个镜像coda怎么没有运行起来呢,怎么检查cuda啊?c4dbf16fa72efd87449795d993c1915a.png 另外 >>> tokenizer = AutoTokenizer.from_pretrained(model_dir, trust_remote_code=True, use_fast=False)

model = AutoModelForCausalLM.from_pretrained(model_dir, device_map="cuda:0", torch_dtype=torch.bfloat16, trust_remote_code=True)
model = model.eval(),这个怎么弄?

展开
收起
陈幕白 2023-12-13 16:39:41 55 0
1 条回答
写回答
取消 提交回答
  • import torch 之后 torch.cuda.is_available() ,显卡型号是比较早的吗,是的话可能不支持bf16,脚本最前面加一下os.environ['CUDA_VISIBLE_DEVICES']='0'明星。此回答整理自钉群 “魔搭ModelScope开发者联盟群 ①”

    2023-12-13 17:04:05
    赞同 展开评论 打赏

包含图像分类、图像生成、人体人脸识别、动作识别、目标分割、视频生成、卡通画、视觉评价、三维视觉等多个领域

相关电子书

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