python安裝TensorFlow吐血整理

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

需要:
系統(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ū)動。

image.png

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

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容