Category: tensorflow

tensorflow禁用GPU 0

tensorflow禁用GPU

Hits: 0如果安裝了tensorflow-gpu 版,但是在測試時只想用 cpu 版本,可以作如下修改: import os os.environ[“CUDA_VISIBLE_DEVICES”]=”-1″ import tensorflow as tf 掃碼關注公衆號: 實用 AI 客棧 獲取最新 AI 資訊與實戰案例 小編微信號 : langu86