最近本人一直在使用CPU版的tensorflow跑神經(jīng)網(wǎng)絡(luò)，CPU的使用率總是飆到100%，風(fēng)扇嗡嗡的轉(zhuǎn)，煩死個人。但看獨立顯卡，使用率卻一直為0%，明顯在摸魚偷懶。而且我也聽說使用GPU跑神經(jīng)網(wǎng)絡(luò)有可能會快一點，于是我決定安裝一個GPU版本的tensorflow。

安裝這個GPU版的tensorflow真是折騰死人了，網(wǎng)上的教程大多都是舊的，有些已經(jīng)不適用了。按照他們教的方法，整的我是云里霧里。最后終于在我自己的摸索下，找到一套可行的安裝流程。

一、前言

在tensorflow英文的官方說明網(wǎng)站中可以看到圖1的提示。這個提示只有英文版有，中文版沒有，真的是坑死了，害我走了好多彎路。

翻譯過來的意思就是：Tensorflow 2.10是最后一個在本地windows上支持GPU的版本。從2.11版本開始，需要在windows WLS2（適用于 Linux 的 Windows 子系統(tǒng)）上安裝才能使用GPU。

所以要在native-windows（一般人用的windows系統(tǒng)）上使用GPU運行tensorflow，就只能安裝2.10版本及以下的版本，或者安裝老版的tensorflow-gpu。

以上的部分內(nèi)容引用自鏈接：https://blog.csdn.net/bragen/article/details/129131278

官方說明鏈接：https://tensorflow.google.cn/install/pip

本人使用的是Python3.9，而且我一般喜歡安裝新版的軟件，所以下面我將介紹在普通windows上可以安裝的最新版本的tensorflow，也就是tensorflow2.10的安裝方法。

二、安裝CUDA

安裝前先打開nvidia控制面板，點左下角的系統(tǒng)信息，再點擊組件，看圖2中圈中的序號，先記住，一會要用。

經(jīng)過我的實驗發(fā)現(xiàn)tensorflow2.10適配的11.x的CUDA。所以我選擇安裝CUDA11的最新版本11.8.0，如圖3所示。

下載鏈接：https://developer.nvidia.com/cuda-toolkit-archive

安裝前請一定確保安裝的CUDA版本低于或等于圖2中圈住的版本號。如不滿足此要求請更新顯卡驅(qū)動或安裝更低版本的CUDA。

下載完后雙擊開始安裝，選擇自定義安裝，只勾選圖4中的這3項即可。

三、安裝cuDNN

下載適配CUDA11版本的cuDNN，我選擇安裝最新版v8.8.1，如圖5所示。

下載鏈接：https://developer.nvidia.com/rdp/cudnn-download

下載完后是一個壓縮包，解壓后直接將bin、include、lib這三個文件夾復(fù)制到CUDA的安裝目錄，如圖6-7所示。

四、添加環(huán)境變量

此電腦→右鍵→屬性→高級系統(tǒng)設(shè)置→環(huán)境變量→系統(tǒng)變量→Path→雙擊→檢查以下路徑是否在Path中，如果沒有，就添加對應(yīng)的路徑，如圖8所示。

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\libnvvp

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\lib

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\include

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\extras\CUPTI\lib64

圖8中環(huán)境變量的路徑均以E:\CUDA\開頭，這是因為我安裝CUDA時使用了自定義安裝路徑。

官方說明鏈接：https://tensorflow.google.cn/install/gpu?hl=zh_cn#windows_setup

五、安裝tensorflow

由于本人沒有安裝虛擬環(huán)境（不推薦，不要學(xué)我），所以我只能介紹在系統(tǒng)直接上安裝tensorflow的方法。

按下win鍵，輸入cmd，在命令提示符處右鍵，選擇以管理員身份運行。

如果以前安裝過其他版本的tensorflow，請在安裝前務(wù)必卸載，可以使用以下命令：

tensorflow從2.x版本后已不再區(qū)分CPU版和GPU版，合為一個包就叫“tensorflow”，所以可以使用以下命令進行安裝：

官方安裝說明：https://tensorflow.google.cn/install/gpu?hl=zh_cn#pip_package

六、驗證

可以使用以下代碼檢驗是否安裝成功：

如果得到的結(jié)果不為0，則證明安裝成功，可以使用GPU進行運算了。

官方說明鏈接：

https://tensorflow.google.cn/guide/gpu?hl=zh_cn#%E8%AE%BE%E7%BD%AE

七、其他

若在安裝中出現(xiàn)缺少dll的問題，可以參考以下鏈接：

https://blog.csdn.net/weixin_44494790/article/details/110254621

但也有可能是CUDA安裝的版本不對，可以根據(jù)缺少的dll名字下劃線后面的數(shù)字來安裝正確版本。

若安裝中還出現(xiàn)了其他問題，可以留言評論（雖然我也不一定能解決??大家可以一起探討），或直接百度查詢。