Windows安裝TensorFlow踩坑

一、判斷顯卡類型

1.1判斷是否是NVIDIA的顯卡。如果是,可以安裝tensorflow-gpu版本。否則,只能安裝tensorflow-cpu版本(版本選擇最終在下載路徑上體現(xiàn))
參考文章https://blog.csdn.net/Candy_GL/article/details/79435151

1.2安裝tensorflow-gpu版本要先更新顯卡驅(qū)動(dòng),安裝Cuda Toolkit和cuDNN。Cuda Toolkit和cuDNN的版本要根據(jù)顯卡版本選擇。
參考文章
版本兼容https://www.cnblogs.com/liaohuiqiang/archive/2018/10/15/9791365.html
驅(qū)動(dòng)安裝http://www.itdecent.cn/p/4c0c66a07c1f
驅(qū)動(dòng)安裝https://blog.csdn.net/u010618587/article/details/82940528

二、安裝Python

2.1目前TensorFlow僅支持到Python3.5。建議使用Anaconda做Python的版本兼容,以及下載TensorFlow
參考文章http://www.itdecent.cn/p/6a1d9522b7c6

三、下載TensorFlow

3.1建議使用國內(nèi)鏡像服務(wù)站下載TensorFlow
參考文章:清華鏡像站:
tensorflow-cpu版本:https://mirrors.tuna.tsinghua.edu.cn/tensorflow/windows/cpu/
tensorflow-gpu版本:https://mirrors.tuna.tsinghua.edu.cn/tensorflow/windows/gpu/

參考命令如下:


image.png

注意:
1.在Anaconda環(huán)境中執(zhí)行下載命令;
2.建議從低版本開始安裝,比如從tensorflow-1.0.0開始安裝,然后慢慢更新到高版本。

四、測試tensorflow環(huán)境

python

import tensorflow
hello = tensorflow.constant("Hello,TensorFlow!")
sess = tensorflow.Session()
print(sess.run(hello))

b'Hello,TensorFlow!'

a = tensorflow.constant(10)
b = tensorflow.constantt(22)
print(sess.run(a + b))

32

五、將Anaconda的環(huán)境與PyCharm關(guān)聯(lián)

image.png

參考文章http://www.itdecent.cn/p/eaee1fadc1e9

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

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

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