Windows10下安裝Tensorflow-GPU你可以避免的坑

Windows10下安裝Tensorflow-GPU你可以避免的坑

工慾善其事必先利其器。有人說,tensorflow-gpu安裝好煩躁哦,各種兼容性問題奧,怎麼搞?

問題 1

如何查看自己的CUDA驅動適配版本?

回答

桌面右鍵打開英偉達控制面板，點擊幫助->系統信息->組件。

問題2

我該下載哪個版本的CUDA和cudnn呢?

回答

問題3

我就頭鐵,用python3.6安裝tensorflow-gpu 1.14.0版本,這樣會報錯嗎?

回答

會。會斬釘截鐵的告訴你缺少XX模塊。

問題 4

我該如何測試我的CUDA系列環境安裝成功了?

回答

問題5

我的電腦有點牛鼻,可以降級CUDA版本嗎?

回答

可以。講道理,是可以的。然後在tf-gpu版本還沒跟上cuda時候,還是得降級,有時候還是有必要的。

問題6:

電腦正常安裝tensorflow-gpu,但是我應該怎麼去測試安裝成功呢?

回答

import tensorflow as tf
import os

os.environ['TF_CPP_MIN_LOG_LEVEL']='2'

a = tf.constant(1.)
b = tf.constant(2.)


with tf.Session() as sess:
    print(sess.run([a,b]))

print(a+b)

print('GPU:', tf.test.is_gpu_available())

然後出現如下結果,說明安裝時正常的。

問題7:

我想臨時快速的下載一個包,怎麼搞?

回答:

pip install keras -i http://pypi.douban.com/simple --trusted-host pypi.douban.com

鏡像源和package可以按照需要改動。

查找靈感

參考1
參考2
參考3
參考4
參考5
參考6