ModelScope中，CUDA_VISIBLE_DEVICES=0 指定0卡不支持这个入参怎么办？
问题一：ModelScope中，CUDA_VISIBLE_DEVICES=0 指定0卡不支持这个入参怎么办？
问题二;我是2.0.1+cu118 cuda是12.1
import torch放在放在os.environ['CUDA_VISIBLE_DEVICES']='0'之后的了
展开
收起
1
条回答
写回答
写回答
-
针对问题一的回答：这种方式您试一下
import os
os.environ['CUDA_VISIBLE_DEVICE']='0'针对问题二的回答;device_map还是"auto"，您试一下，我这边单卡可以，多卡没环境。您的代码中设置os.environ之后用device_count查看一下。2，3行放最前面。——此回答整理自钉群：魔搭ModelScope开发者联盟群 ①2023-11-08 22:02:57赞同 展开评论 打赏
相关问答
热门讨论
热门文章
ModelScope中，模型下载默认路径在哪个路径？
7904
com/action/joingroup?code=v1是什么意思
6235
请问在 ModelScope 上的模型断网使用报这个错误啥原因了?
1868
服务器上onnxruntime-gpu 调用结束，如何释放显存
4727
如何下载modelscope模型？
1063
ModelScope中，大佬们4卡跑lora时，遇到这样的错，怎么解决？
1635
我希望通过damo-YOLO训练1500*1500的图片
8199
ModelScope 使用14B模型推理时报这个错误？
1608
ttsfrd 如何在windows安装
767
请问modelscope中，有一个依赖，kantts，要怎么安装？
1780
展开全部
YOLOv10实战：SPPF原创自研 | SPPF_attention，重新设计加入注意力机制 | NEU-DET为案列进行展开
360
关于flux.1 loras的8个问题
231
YOLOv10实战：红外小目标实战 | 多头检测器提升小目标检测精度
628
关键点检测从入门到进阶
1141
Stable Diffusion 介绍与入门
1519
一种基于YOLOv8改进的高精度红外小目标检测算法 （原创自研）
298
人像抠图
961
人像增强修复
831
达摩院OpenVI-视频目标跟踪ICASSP 2023 Oral 渐进式上下文Transformer跟踪器 ProContEXT
683
ModelScope模型使用与EAS部署调用
191
展开全部
