1. 程式人生 > >VS2017 + CUDA 9.2 + CUDNN 7.1.4 + tensorflow 安裝

VS2017 + CUDA 9.2 + CUDNN 7.1.4 + tensorflow 安裝

h5py with floating blank col 進行 ima inf height

參照步驟: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 安裝