python安装TensorFlow吐血整理

2019-01-22  本文已影响7人  续袁

1.安装特定版本的TensorFlow

(1)pip install tensorflow-gpu==1.9.0
(2)pip install tensorflow-gpu
必须用命令(1)才能安装想要的新版本,命令(2)只会安装TensorFlow1.21
安装anaconda,然后用python的pip可以安装特定版本的tensorflow,如:

pip install tensorflow-gpu==1.4.0
pip install tensorflow==1.8.0
tensorFlow安装过程

2.安装特定版本的TensorFlow-gpu

需要:
系统环境:windows10 64位
vs2015或vs2017

显卡:GTX1060

CUDA版本:cuda_9.0.176_win10(配套的cudnn一定要是7版本)

下载地址:https://developer.nvidia.com/cuda-90-download-archive

CUDNN版本:cudnn-9.0-windows10-x64-v7(7版本的)

下载地址:https://developer.nvidia.com/cudnn

TensorFlow-gpu版本:TensorFlow1.7(1.6和1.5也支持)

1.下载CUDA, https://developer.nvidia.com/cuda-90-download-archive

按照自己机器的信息选择,下图是我自己机器信息:

下载完之后直接点击运行,会自行安装,安装时会让你选择一个文件夹作为抽取文件临时存放,记住这不是安装文件夹,安装文件夹系统会自动安装到C盘,具体我后面会提到,临时文件夹安装完后会自动删除。

如果出现安装不成功的情况,则需要卸载系统中已有的显卡驱动。

image.png

3.卸载特定版本的TensorFlow-gpu

卸载旧版本
pip uninstall tensorflow
conda uninstall numpy
pip list 列出当前环境所安装的包

参考资料

[1] tensorflow指定版本的安装及升级到最新版
[2] Win10 NVDIA CUDA安装失败解决方案
[3] Win10环境+ CUDA9.0 +CUDNN7.0+TensorFlow1.7/1.6/1.5配置

[1] cuDNN Download

上一篇 下一篇

猜你喜欢

热点阅读