TensorFlow版本升降级/解决GPU与CUDA版本冲突

目前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

查看当前tensorflow版本:

python
import tensorflow as tf
tf.__version__

查询tensorflow安装路径为:

tf.__path__


Ref:

https://blog.csdn.net/junmuzi/article/details/78357371




你可能感兴趣的:(TensorFlow)