ModelScope中,CUDA_VISIBLE_DEVICES=0 指定0卡不支持这个入参怎么办?

问题一:ModelScope中,CUDA_VISIBLE_DEVICES=0 指定0卡不支持这个入参怎么办?lQLPJxhZhlo4Y3jNARvNCLqwDjvyq5RA4gsFNkvLbYAUAA_2234_283.png问题二;我是2.0.1+cu118 cuda是12.1
lQLPJx7HtbycfujNBVHNCv-wasvtJzIeGGgFNoX0ywC4AA_2815_1361.pngimport torch放在放在os.environ['CUDA_VISIBLE_DEVICES']='0'之后的了
lQLPJxzBTEV3J1jNAkHNBvOwhEuwWmVOOKAFNpBMNkCjAA_1779_577.png

展开
收起
多麻辣哦 2023-11-08 21:01:33 302 分享 版权
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
    赞同 展开评论

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

热门讨论

热门文章

还有其他疑问?
咨询AI助理