windows 10 Tensorflow GPU 版本安装

时隔一年,再装tensorflow,发现windows10上面安装已经比之前简单了很多

https://www.tensorflow.org/install/gpu?hl=zh-cn
现在GPU 版本和 CPU版本已经合并了,所以可以直接安装
pip install tensorflow

碰到错误,可能是因为之前装 tensorflow-gpu失败导致的
https://www.cnblogs.com/conver/p/11141176.html

Cannot uninstall ‘wrapt’. It is a distutils installed project
https://www.cnblogs.com/conver/p/11141176.html

直接:
pip install -U --ignore-installed wrapt enum34 simplejson netaddr

然后装CUDA 和 cudnn
前者下载exe,一直点下一步,如果要自己选安装目录,选择自定义安装
然后把 bin目录路径添加到系统PATH环境变量中,貌似需要重启实现环境变量生效

后者的安装则是拷贝cudnn中的文件到对应的cuda安装目录中即可,主要是版本问题,
对应cuda10.1 , cudnn版本不对不会影响 python中 import tensorflow,但运行时会报错,试了一些早期版本发现不行, 最后用了 cudnn-10.1-windows10-x64-v8.0.5.39 才成功,

上一篇:Openpose安装


下一篇:Windows安装NVIDIA驱动&&CUDA&&cuDNN&&Anaconda步骤