Пользовательское расширение C ++: torch / torch.h не найдено

Question

Я следую этому руководству , чтобы создать расширение C ++ для Pytorch. Мой код C ++ дает следующую ошибку:

test.cpp:3:10: fatal error: torch/torch.h: No such file or directory
 #include <torch/torch.h>

Как получить torch.h заголовочный файл? Есть ли какая-нибудь pytorch-dev версия?

Answer 1

Скорее всего, вам нужно установить версию факела на С ++.

Если вы зайдете на сайт www.pytorch.org, он подскажет вам, какая загрузка вам подходит. Просто убедитесь, что выбрали версию C ++.

Надеюсь, это работает!