tensorflow gpu安裝的一些注意

2021-09-08 05:25:56 字數 670 閱讀 2077

按正確的順序安裝,嚴格安裝特定的版本

1.1,安裝9.0 patch1之前,先刪除掉c:\program files\nvidia corporation\installer2,否則閃退。

1.2,安裝cudnn後,檢查下%path%裡是否有c:\program files\nvidia gpu computing toolkit\cuda\v9.0\bin和libnvvp。

以及%cuda_path%=c:\program files\nvidia gpu computing toolkit\cuda\v9.0

以及%cuda_path_v9_0%=c:\program files\nvidia gpu computing toolkit\cuda\v9.0

2,安裝顯示卡驅動,這個可以裝最新的。

3,pip install tensorflow-gpu。注意不要安裝非「-gpu」的cpu版。如果安了,倆全解除安裝,再單安-gpu。

3,pip uninstall protobuf, pip install protobuf==3.6.0(嚴格和你的python版本相同,3.6.1就不幹活)

4,進python驗證一下,import tensorflow as tf; sess=tf.session(config=tf.configproto(log_device_placement=true));

離線安裝tensorflow gpu

前言 安裝前先確定電腦可以使用gpu執行tensorflow,否則浪費時間。nvidia 英偉達 官網官網檢視顯示卡是否支援gpu執行,compute capability需要在3.0及以上,否則安裝後,不能用gpu執行程式,會自動呼叫cpu執行。nvidia官網 密碼 0pmq 檔案列表 anac...

Tensorflow gpu版本安裝

這裡安裝想要嘗試最新的cuda10.1,但是失敗了,得到的教訓就是要嚴格的按照官方的版本要求。因為tensorflow1.13是基於cuda10的並沒有整合10.1,個人推測,兩家是不同的公司cuda10.1正常不會想到要相容自己的上層應用,或者s是tf直接寫死的需要cuda10。不管怎樣,現在總結...

tensorflow gpu版本安裝

1.tensorflow gpu 1.15.0版本相容無gpu的pc 執行import tensorflow時,自動判斷有無gpu或者當前gpu是否支援tensorflow,若無gpu或當前gpu不支援tensorflow會自動使用cpu 2.tensorflow gpu 版本小於1.15.0時只能...