tensorflow 在windows 下怎么使用gpu

2024-11-15 10:49:23
推荐回答(2个)
回答1:

安装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计算

回答2:

通过浏览器在线直接直接调用tensorflow环境,利用cuda编程查看GPU的型号信息