我的AI之路(47)--使用自己的數據集訓練EfficientDet

       訓練用的服務器的芯片一般都是X86架構的,所以可以直接從hub.docker.com上拉個安裝了pytorch的鏡像下來作爲訓練用的環境,省去了很多安裝步驟,例如拉取pytorch/pytorch:1.2-cuda10.0-cudnn7-devel(注意一般不要使用runtime鏡像,runtime鏡像裏面只安裝了最小的支持環境,很多工具都沒安裝,devel鏡像一般是比較全面的):
相關文章
相關標籤/搜索