1. 程式人生 > >tensorflow使用gpu進行計算的配置步驟

tensorflow使用gpu進行計算的配置步驟


2、安裝cudnn https://developer.nvidia.com/cudnn


3、pip install tensorflow-gpu,如果只需要cpu版本則pip install tensorflow