当我这样操作时,传入device为0或者“cuda:0”,都无法让其运行到指定gpuself.skin_retouching = pipeline(Tasks.skin_retouching, model='damo/cv_unet_skin-retouching',device=device)
默认直接加载到多张卡上。
而且我发现对于不同的damo模型,使用这样的device传参方法,有的还会报错?难道达摩院内部的接口不统一吗?
对于我的需求我该如何修改我的代码?
当我这样操作时,传入device为0或者“cuda:0”,都无法让其运行到指定gpuself.skin_retouching = pipeline(Tasks.skin_retouching, model='damo/cv_unet_skin-retouching',device=device)
默认直接加载到多张卡上。
而且我发现对于不同的damo模型,使用这样的device传参方法,有的还会报错?难道达摩院内部的接口不统一吗?
对于我的需求我该如何修改我的代码?