猿问
tensorflow 怎样确定是cpu 还是 gpu?
tensorflow 怎样确定是cpu 还是 gpu
米琪卡哇伊
浏览 2469
回答 1
1回答
尚方宝剑之说
安装GPU版本的tensorflow就可以了pip install --upgrade 还需要安装GPU加速包,下载 CUDA 和 cuDNN 的安装包了,注意版本号分别是 CUDA 8.0 和 cuDNN 5.1。如果都安装成功,运行程序时会自动使用GPU计算
0
0
0
随时随地看视频
慕课网APP
相关分类
人工智能
我要回答