首页 > 其他 > 详细

VS2017 + CUDA 9.2 + CUDNN 7.1.4 + tensorflow 安装

时间:2018-07-26 14:00:13      阅读:241      评论:0      收藏:0      [点我收藏+]

参照步骤:https://blog.csdn.net/weixin_42359147/article/details/80622306

坑:安装完成后

import tensorflow

出现:“FutureWarning: Conversion of the second argument of issubdtype from `float` to `np.floating` is deprecated. In future, it will be treated as `np.float64 == np.dtype(float).type`.”错误。

解决方法: 对h5py进行更新升级

pip3 install h5py==2.8.0rc1

 测试:

import tensorflow as tf
a = tf.test.is_built_with_cuda()
b = tf.test.is_gpu_available(cuda_only = False, min_cuda_compute_capability = None)
print(a,b)

技术分享图片

 

VS2017 + CUDA 9.2 + CUDNN 7.1.4 + tensorflow 安装

原文:https://www.cnblogs.com/Xlni/p/9371022.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!