目前tensorflow官方提供的版本只支持cuda 9.0,非源码安装要求GPU计算能力在3.5以上。
注意官方提供的安装方式将默认安装最新的TensorFlow稳定版,即便安装cuda 8.0也无法运行
C:\> pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple --upgrade tensorflow-gpu
对于GPU比较老的机器,有两个方法解决
1)从源码编译,源码编译不支持windows,不过文档似乎提供了强行编译的解决办法
2)降低TensorFlow版本,1.4.0版本(cuda 8.0)支持3.0的GPU
pip install tensorflow-gpu==1.4.0
注意,升级python的package要注意,很有可能有版本冲突,没事不建议升级
——————————————————————
升级TensorFlow:
pip install --upgrade tensorflow-gpu
pip install--upgrade tensorflow