tensorflow gpu 安裝填坑記錄

2021-09-19 08:13:12 字數 1110 閱讀 7462

anacoda

windows10 64作業系統專業版

1050ti顯示卡

顯示卡驅動和cuda安裝一路全選預設即可

cudnn 的安裝

#建立name為tensorflow的虛擬環境,其中python版本為3.6

conda create -n tensorflow python=3.6

#在當前cmd視窗環境切換到 tensorflow蓄力環境

activate tensorflow

#安裝tensorflow

pip install tensorflow-gpu

#測試python -c "import tensorflow as tf; tf.enable_eager_execution(); print(tf.reduce_sum(tf.random_normal([1000, 1000])))"

官網參考:

#切換環境

activate tensorflow

#安裝ipykernel

pip install ipykernel

#將當前環境作為乙個新的jupyter的kernel

python -m ipykernel install --name tensorflow

完成之後開啟jupyter notebook 就可以看到我們剛剛新建的kernel,使用這個kernel時,就是使用了tensorflow的虛擬環境

#在base環境下操作

安裝jupyter_contrib_nbextensions

pip install jupyter_contrib_nbextensions

執行jupyter contrib nbextension install --user

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