tensorflow-gpu 2.0.0a0 windows踩坑记录

win10环境下,按照CUDA-CUDNN-tensorflow的安装顺序

1.tensorflow DLL load failed: 找不到指定的模块。

CUDA和CUDNN版本不对

CUDA10.1和对应的CUDNN不支持当前tensorflow2.0.0a0,切换为10.0的CUDA就行

CUDA10.0: CUDA Toolkit 10.0 Archive

对应的CUDNN: https://developer.nvidia.com/rdp/cudnn-download

另外,配置CUDA的时候还增加了两个环境变量,

PATH下的

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\extras\CUPTI\libx64

PS:下载CUDA的时候,local版本下载失败,到80%左右就会强行报错

解决方案:选择network版的在线下载安装就行

2.ModuleNotFoundError: No module named ‘numpy.core._multiarray_umath’

numpy版本太低,升级一下版本即可

    原文作者:桥风
    原文地址: https://zhuanlan.zhihu.com/p/63600226
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞