Skip to content

AssertionError: Torch not compiled with CUDA enabled #536

Open
@MalteEbner

Description

@MalteEbner

When running the code without a CUDA Gpu and/or pytorch not compiled with it, the following error occurs: "AssertionError: Torch not compiled with CUDA enabled". The reason is that "ssd_pytorch/layers/modules/multibox_loss.py:35 sets the argument use_gpu per default to true. This can be easily fixed, see #537.

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

      Development

      Participants

      @MalteEbner

      Issue actions

        AssertionError: Torch not compiled with CUDA enabled · Issue #536 · amdegroot/ssd.pytorch