Window环境下使用显卡 GPU 加速深度学习并与 CPU 进行对比(配置过程+实例)
1 安装 tensorflow-gpupip install tensorflow-gpu测试是否安装成功import tensorflow as tfprint(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 ·
1182 阅读 ·
0 评论