Tensorflow GPU安裝以及CUDA

2021-10-07 11:01:54 字數 1103 閱讀 8384

python:3.7.1

tensorflow-gpu:1.14.0

cuda:10.0.132

cudnn:7.4.1.5

顯示卡:gtx1660ti

python版本要求沒那麼嚴格,別過低就好了。盡量裝個折中的,不新不老。

唯一難點在於pip.exe沒有新增到系統路徑。導致pip install 指令報錯

「pip' 不是內部或外部命令,也不是可執行的程式或批處理檔案」的報錯。
解決辦法:python新增環境變數

win+r,輸入cmd。

pip install tensorflow-gpu==1.14.0  -i

cuda安裝教程,巨詳細

測試tensorflow-gpu是否能正常使用的官網測試**

import tensorflow as tf

mnist = tf.keras.datasets.mnist

(x_train, y_train),(x_test, y_test) = mnist.load_data()

x_train, x_test = x_train / 255.0, x_test / 255.0

model = tf.keras.models.sequential([

tf.keras.layers.flatten(),

tf.keras.layers.dense(512, activation=tf.nn.relu),

tf.keras.layers.dropout(0.2),

tf.keras.layers.dense(10, activation=tf.nn.softmax)

])model.compile(optimizer='adam',

loss='sparse_categorical_crossentropy',

metrics=['accuracy'])

model.fit(x_train, y_train, epochs=5)

model.evaluate(x_test, y_test)

print(y_test)

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