请问测试可以分显卡跑吗？

hustvl / ViTMatte

[Information Fusion (Vol.103, Mar. '24)] Boosting Image Matting with Pretrained Plain Vision Transformers

MIT License

339 stars 33 forks source link

Closed davislee546 closed 1 year ago

davislee546 commented 1 year ago

默认只跑一张卡，跑着跑着就会报错"cuda out of memory"

JingfengYao commented 1 year ago

目前是不行的。你可以参考issue 10来尝试降低你测试的显存。

davislee546 commented 1 year ago

目前是不行的。你可以参考issue 10来尝试降低你测试的显存。

感谢！确实可以！