前两天新安装的win10在昨晚升级过后TensorFlow就开始失效,于是重新安装环境,今天pip 安装TensorFlow的时候已经是1.5.0版本,前两天还是1.4.1呢。安装完毕之后import tensorflow总是报错,具体报的啥做不记得了,大概是找不到指定的模块。试了CUDA8.0、CUDA9.1,修改了cudnn文件名都不行,然后Google一下发现是tensorflow1.5.0版本只支持cuda9.0,然而NVIDIA中国官网上并没有CUDA9.0的下载链接,英文官网上有的,这里贴上链接,有需要的朋友可以直接下载:https://developer.nvidia.com/cuda-90-download-archive
网友评论