时隔一年,再装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 才成功,