在 Windows 命令提示符下從源構建 pytorch 時啟用 CAFFE2

Question

所以，我在 Windows 平台上使用 Yolo7 進行了 model 訓練，並且

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. 下面的錯誤提示怎么辦？ 如何在我的 Windows 上構建 pytorch 時啟用BUILD_CAFFE2=1 ？ 當然不使用 Conda。 僅在我的 Windows 命令提示符下。
2. 我使用以下來源安裝了 pytorch https://github.com/pytorch/pytorch#from-source使用此來源的命令安裝 caffe2 https://caffe2.ai/docs/getting-started.html?platform=windows&configuration=編譯但是在訓練我的 model 時仍然顯示以下錯誤。

我只需要知道在 windows 命令提示符下啟用build_caffe2=1的命令。

    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)

   
 

Answer 1

我已經解決了這個問題，但是在安裝 pytorch 之前在命令提示符下設置 BUILD_CAFFE2=1，代碼如下。

set BUILD_CAFFE2=1