Window环境下使用显卡 GPU 加速深度学习并与 CPU 进行对比（配置过程+实例）
1 安装 tensorflow-gpu
pip install tensorflow-gpu
测试是否安装成功
import tensorflow as tf
print(tf.__version__)
hello=tf.constant('hello world')
sess=tf.compat.v1.Session()
print(hello.numpy())
2 下载 文件
下载前，先确定自己的 tensorflow 版本，然后在确定 cuda 和 cudnn 版本
具体对应关系如下：
Tens
原创
2020-09-12 18:24:35 ·
1138 阅读 ·
0 评论
关注
分享
扫一扫