網站架設
- from ultralytics import YOLO
-
- # Load a model
- model = YOLO("yolov8n.yaml") # build a new model from scratch
- model = YOLO("yolov8n.pt") # load a pretrained model (recommended for training)
-
- if __name__ == '__main__':
- # Use the model
- model.train(data="coco128.yaml", epochs=3) # train the model
- metrics = model.val() # evaluate model performance on the validation set
- results = model("https://ultralytics.com/images/bus.jpg") # predict on an image
- path = model.export(format="onnx") # export the model to ONNX format
複製代碼
近從新調試一段pytorch 程式碼，以前的伺服器上完全沒問題，但換了一台機器，從頭安裝了新版本的cuda， anaconda ，pytorch 等，之前的程式碼出現各種版本不合適的問題。
問題：
此刻說說這個問題。運行pytorch 時泛起的環境如下：
- RuntimeError:
- An attempt has been made to start a new process before the
- current process has finished its bootstrapping phase.
-
- This probably means that you are not using fork to start your
- child processes and you have forgotten to use the proper idiom
- in the main module:
-
- if __name__ == '__main__':
- freeze_support()
- ...
-
- The "freeze_support()" line can be omitted if the program
- is not going to be frozen to produce an executable.
複製代碼
試過，有效，一次通過。
2. 不利用多線程，也就是去掉num_workers 參數，或設定 num_workers=0。
（目前還沒試過，因為我照舊想用多履行緒。）
文章出處： NetYea 網頁設計
參考文章
https://blog.csdn.net/u014546828/article/details/109235539
原因：
在網上查找了局，緣由是多進程的緣由。具體可參考博文：Python 中的if __name__ == '__main__' 該若何理解。
解決方式：
既然是多履行緒的原因，那麼可以從兩個角度解決問題：
1. 程式碼在運行 epoch 之前，加上 if __name__=='__main__'
