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.png3.卸载特定版本的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
网友评论