深度學習模型函數執行完畢後顯存不釋放問題解決方案

使用多線程的方法來解決 以下定義model_predict()函數爲調用深度學習模型預測結果函數，返回接口列表，使用多線程的方法可使得調用函數後釋放顯卡，以便在同張顯卡上調用第二個深度學習模型。 具體代碼以下python import multiprocessing # return_dict用於函數返回值 def model_predict(X_test, return_dict): ...

>>阅读原文<<