时隔一年,再装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
CUDA: cuda_10.1.105_418.96_win10.exe
CUDNN: cudnn-10.1-windows10-x64-v7.6.5.32
前者下载exe,一直点下一步,如果要自己选安装目录,选择自定义安装
然后把 bin目录路径添加到系统PATH环境变量中,貌似需要重启实现环境变量生效
后者的安装则是拷贝cudnn中的文件到对应的cuda安装目录中即可,主要是版本问题,
对应cuda10.1 , cudnn版本不对不会影响 python中 import tensorflow,但运行时会报错,试了一些早期版本发现不行, 最后用了 cudnn-10.1-windows10-x64-v7.6.5.32才成功,
检验;
import tensorflow as tf
print(tf.__version__)
print(tf.test.is_gpu_available())
tensorflow版本为2.3.1