Tensorflow gpu安裝過程記錄

2021-08-27 05:02:24 字數 914 閱讀 7288

install nvidia accelerated graphics driver for linux-x86_64 *** xx?

選項中,輸入「n」 表示不安裝,否則安裝不成功,我在此反覆多次,最後在autocyz的博文幫助下才成功。感謝autocyz,參見:ubuntu16.04安裝 cuda7.5(**:

3. 安裝cuda和cudnn要把它們新增到環境變數,可查閱度娘。我是參考gtx_viewer的博文「nvidia cuda toolkit的安裝(cuda)」(見:

sudo gedit /etc/profile

然後,在開啟的檔案最後新增(此處cuda-8.0是我cuda的安裝路徑,/home/weizhong/software/cuda表示cudnn解壓縮的路徑):

export path=/usr/local/cuda-8.0/bin:$path

export ld_library_path=/usr/local/cuda-8.0/lib64:$ld_library_path

export ld_library_path=/home/weizhong/software/cuda/lib64:$ld_library_path

再新增lib庫路徑:

sudo vim /etc/ld.so.conf.d/cuda.conf

內容如下:

/usr/local/cuda-8.0/lib64

/home/weizhong/software/cuda/lib64

/cudnn_dir/lib64

/lib

/usr/lib

/usr/lib32

4. 建立conda環境很重要,能省不少麻煩事,然後在建立的環境下安裝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時只能...