keras強制使用cpu

安裝了gpu版本後,keras預測也使用了GPU。因爲項目要求測試CPU下速度,所以,在python調用代碼中,作以下限制:python import os os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
相關文章
相關標籤/搜索