学习笔记4.3.AI相关-crypten使用gpu运行卷积conv时出现AssertionError: more than one group is unsupported on GPU问题的解决

Adongua

于 2024-01-18 22:23:06 发布

阅读量584

点赞数 8

分类专栏： # Data security 文章标签：人工智能学习笔记隐私保护

本文链接：https://blog.csdn.net/qq_51352378/article/details/135685485

版权

这篇文章介绍了在使用Crypten库进行AI相关任务时，遇到在GPU上运行卷积conv操作时的AssertionError，解决方法是针对特定问题修改了cuda_tensor.py文件中的__patched_conv_ops方法，移除或处理groups参数以适应GPU环境。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

文章目录

学习笔记4.3.AI相关-crypten使用gpu运行卷积conv时出现AssertionError: more than one group is unsupported on GPU问题的解决

学习笔记4.3.AI相关-crypten使用gpu运行卷积conv时出现AssertionError: more than one group is unsupported on GPU问题的解决

参考链接
https://github.com/facebookresearch/CrypTen/issues/386

解决方法
找到crypten/cuda/cuda_tensor.py中的__patched_conv_ops方法，做如下修改：

    @staticmethod
    def __patched_conv_ops(op, x, y, *args, **kwargs):
        if "groups" in kwargs:
            groups = kwargs["groups"]
            #del kwargs["groups"]
        else:
            groups = 1
        bs, c, *img

最低0.47元/天解锁文章