tensorflow 在windows 下怎么使用gpu
2个回答
展开全部
安装GPU版本的tensorflow就可以了pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.0rc0-cp35-cp35m-win_amd64.whl
还需要安装GPU加速包,下载 CUDA 和 cuDNN 的安装包了,注意版本号分别是 CUDA 8.0 和 cuDNN 5.1。
如果都安装成功,运行程序时会自动使用GPU计算
还需要安装GPU加速包,下载 CUDA 和 cuDNN 的安装包了,注意版本号分别是 CUDA 8.0 和 cuDNN 5.1。
如果都安装成功,运行程序时会自动使用GPU计算
推荐律师服务:
若未解决您的问题,请您详细描述您的问题,通过百度律临进行免费专业咨询