Включение CAFFE2 при сборке pytorch из исходного кода в командной строке Windows

Итак, я делал модельный поезд, используя Yolo7 на платформе Windows и

C:\Users\LENOVO>python train.py --weights yolov7.pt --data "data/custom.yaml" --workers 4 --batch-size 4 --img 416 --cfg cfg/training/yolov7. yaml --name yolov7 --hyp data/hyp.scratch.p5.yaml

После выполнения вышеуказанной команды в моей командной строке в Windows появилась следующая трассировка стека ошибки. Мой вопрос:

  1. Как сделать предложения ошибки ниже? Как включить BUILD_CAFFE2=1 при сборке pytorch в моей Windows? Не используя Conda, конечно. Только в моей командной строке Windows.
  2. Я установил pytorch из следующего источника https://github.com/pytorch/pytorch#from-source Установите caffe2 с помощью команд из этого источника https://caffe2.ai/docs/getting-started.html?platform=windows&configuration=compile Но следующая ошибка все еще отображается при обучении моей модели.

Мне просто нужно знать команду включения build_caffe2=1 в командной строке Windows.

    C:\Users\LENOVO\AppData\Local\Programs\Python\Python310\lib\site-packages\caffe2\__init__.py:5: UserWarning: Caffe2 support is not fully enabled in this PyTorch build. Please enable Caffe2 by building PyTorch from source with `BUILD_CAFFE2=1` flag.
     warnings.warn("Caffe2 support is not fully enabled in this PyTorch build. "
    C:\Users\LENOVO\AppData\Local\Programs\Python\Python310\lib\site-packages\caffe2\proto\__init__.py:17: UserWarning: Caffe2 support is not enabled in this PyTorch build. Please enable Caffe2 by building PyTorch from source with `BUILD_CAFFE2=1` flag.
     warnings.warn('Caffe2 support is not enabled in this PyTorch build. '
    C:\Users\LENOVO\AppData\Local\Programs\Python\Python310\lib\site-packages\caffe2\python\__init__.py:9: UserWarning: Caffe2 support is not enabled in this PyTorch build. Please enable Caffe2 by building PyTorch from source with `BUILD_CAFFE2=1` flag.
     warnings.warn('Caffe2 support is not enabled in this PyTorch build. '
    Traceback (most recent call last):
     File "C:\Users\LENOVO\train.py", line 8, in <module>
    from caffe2.python import core, scope
      File "C:\Users\LENOVO\AppData\Local\Programs\Python\Python310\lib\site-packages\caffe2\python\__init__.py", line 7, in <module>
    from caffe2.proto import caffe2_pb2
      File "C:\Users\LENOVO\AppData\Local\Programs\Python\Python310\lib\site-packages\caffe2\proto\__init__.py", line 15, in <module>
    from caffe2.proto import caffe2_pb2, metanet_pb2, torch_pb2
    ImportError: cannot import name 'metanet_pb2' from partially initialized module 'caffe2.proto' (most likely due to a circular import) (C:\Users\LENOVO\AppData\Local\Programs\Python\Python310\lib\site-packages\caffe2\proto\__init__.py)

   
 
Мутабельность и переработка объектов в Python
Мутабельность и переработка объектов в Python
Объекты являются основной конструкцией любого языка ООП, и каждый язык определяет свой собственный синтаксис для их создания, обновления и...
Другой маршрут в Flask Python
Другой маршрут в Flask Python
Flask - это фреймворк, который поддерживает веб-приложения. В этой статье я покажу, как мы можем использовать @app .route в flask, чтобы иметь другую...
14 Задание: Типы данных и структуры данных Python для DevOps
14 Задание: Типы данных и структуры данных Python для DevOps
Проверить тип данных используемой переменной, мы можем просто написать: your_variable=100
Python PyPDF2 - запись метаданных PDF
Python PyPDF2 - запись метаданных PDF
Python скрипт, который будет записывать метаданные в PDF файл, для этого мы будем использовать PDF ридер из библиотеки PyPDF2 . PyPDF2 - это...
Переменные, типы данных и операторы в Python
Переменные, типы данных и операторы в Python
В Python переменные используются как место для хранения значений. Пример переменной формы:
Почему Python - идеальный выбор для проекта AI и ML
Почему Python - идеальный выбор для проекта AI и ML
Блог, которым поделился Harikrishna Kundariya в нашем сообществе Developer Nation Community.
0
0
181
1
Перейти к ответу Данный вопрос помечен как решенный

Ответы 1

Ответ принят как подходящий

Я решил проблему, но установил BUILD_CAFFE2=1 в командной строке перед установкой pytorch со следующим кодом.

set BUILD_CAFFE2=1

Другие вопросы по теме