tensorflow gpu安裝過程記錄及心得

2021-10-07 03:18:24 字數 1002 閱讀 4982

首先講一下我的電腦配置

windows7 64位

顯示卡 gtx 1660 6g

python3.7

cuda10.0

cudnn7.6

tensorflow-gpu1.13.1

pycharm2018(這個版本可以破解到2023年,後面的版本貌似不行)

因為之前用的是tensorflow-cpu,所以要先解除安裝掉cpu版本的tensorflow

pip uninstall tensorflow
之後檢驗cuda是否安裝成功:開啟cmd輸入:

nvcc -v
回車就可以檢視cuda版本

這個樣子就是安裝成功了

c:\program files\nvidia gpu computing toolkit\cuda\v10.

0

cudnn安裝成功

之後就是安裝tensorflow-gpu了

.1至此,安裝成功。

檢驗是否成功:

import tensorflow as tf

hello = tf.constant(

'hello,tensorflow!'

)sess = tf.session(

)print

(sess.run(hello)

)

沒有報錯即為成功

以上是我自己安裝tensorflow的過程及理解,如有問題請指出。

離線安裝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時只能...