遵循本教程Tensorflow GPU,我做到了,它运行完美。
注意! -安装 9.0版! Tensorflow-gpu不支持较新的版本
脚步:
- 卸载旧的tensorflow
- 安装tensorflow-gpu
pip install tensorflow-gpu
- 安装Nvidia图形卡和驱动程序(您可能已经拥有)
- 下载并安装CUDA
- 下载并安装cuDNN
- 通过简单程序验证
from tensorflow.python.client import device_libprint(device_lib.list_local_devices())
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)