ubuntu 16.04 使用keras框架 對比cpu和gpu訓練神經網絡的速度

本機配置: cpu:i5-4200H gpu:gtx 950M 照例,先查看本機可以使用的設備 在spyder中輸入: from tensorflow.python.client import device_lib print(device_lib.list_local_devices())  也就是說,我可以選擇調用cpu,也可以選擇調用gpu。 先上測試代碼: import tensorflo
相關文章
相關標籤/搜索