tensorflow 怎样确定是cpu 还是 gpu
1 回答
尚方宝剑之说
TA贡献1788条经验 获得超4个赞
安装GPU版本的tensorflow就可以了pip install --upgrade 还需要安装GPU加速包，下载 CUDA 和 cuDNN 的安装包了，注意版本号分别是 CUDA 8.0 和 cuDNN 5.1。
如果都安装成功，运行程序时会自动使用GPU计算
- 1 回答
- 0 关注
- 2515 浏览
添加回答
举报
0/150
提交
取消