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

2.安裝特定版本的TensorFlow-gpu
需要:
系統(tǒng)環(huán)境: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
按照自己機器的信息選擇,下圖是我自己機器信息:
下載完之后直接點擊運行,會自行安裝,安裝時會讓你選擇一個文件夾作為抽取文件臨時存放,記住這不是安裝文件夾,安裝文件夾系統(tǒng)會自動安裝到C盤,具體我后面會提到,臨時文件夾安裝完后會自動刪除。
如果出現(xiàn)安裝不成功的情況,則需要卸載系統(tǒng)中已有的顯卡驅(qū)動。

3.卸載特定版本的TensorFlow-gpu
卸載舊版本
pip uninstall tensorflow
conda uninstall numpy
pip list 列出當(dāng)前環(huán)境所安裝的包
參考資料
[1] tensorflow指定版本的安裝及升級到最新版
[2] Win10 NVDIA CUDA安裝失敗解決方案
[3] Win10環(huán)境+ CUDA9.0 +CUDNN7.0+TensorFlow1.7/1.6/1.5配置
[1] cuDNN Download