Detectron2: Detectron2 performed on the CPU with a "Torch not compiled with CUDA enabled" error

Created on 13 May 2020  路  1Comment  路  Source: facebookresearch/detectron2

The Issue:

  1. Detectron2 performed on the CPU with a "Torch not compiled with CUDA enabled" error
    Configure Detectron2 environment on a computer without GPU graphics card. After the configuration is OK, run the following code. The weight file is downloaded well in advance and placed in a fixed location.
    code:
    灞忓箷蹇収 2020-05-13 涓嬪崍8 17 13

errors:
灞忓箷蹇収 2020-05-13 涓嬪崍8 16 59
The error is shown below:
灞忓箷蹇収 2020-05-13 涓嬪崍8 04 09

picture sswhqqw

Most helpful comment

Hi, try adding this after cfg.MODEL.WEIGHTS:
cfg.MODEL.DEVICE = 'cpu'

picture ksablauskas on 13 May 2020
👍4 🚀1 ❤1 🎉1

>All comments

Hi, try adding this after cfg.MODEL.WEIGHTS:
cfg.MODEL.DEVICE = 'cpu'

ksablauskas picture ksablauskas on 13 May 2020
👍4 🚀1 ❤1 🎉1
Was this page helpful?
0 / 5 - 0 ratings

Related issues

marcoippolito picture marcoippolito  路  4Comments
AntonBaumannDE picture AntonBaumannDE  路  3Comments
choasup picture choasup  路  3Comments
invisprints picture invisprints  路  4Comments
wytcsuch picture wytcsuch  路  4Comments