目錄

基礎知識信息了解

顯卡算力

CUDA兼容

Tensorflow gpu安裝

CUDA/cuDNN匹配和下載

查看Conda driver的版本

下載CUDA工具包

查看對應cuDNN版本

下載cuDNN加速庫

CUDA/cuDNN安裝

CUDA安裝方法

cuDNN加速庫安裝

配置CUDA/cuDNN環(huán)境變量

配置環(huán)境變量

核驗是否安裝成功

Tensorflow-gpu安裝

命令安裝

報錯處理

核驗安裝結果

---

直通車：人工智能發(fā)展歷程和工具搭建學習-CSDN博客

????????通過之前的文章學習，我們已經(jīng)安裝好了Anaconda和Tensorflow2.4，但是在后期的學習中，會涉及到神經(jīng)網(wǎng)絡的學習等數(shù)據(jù)量較大的操作，普通的tensorflow-cpu版本處理速度較慢，所以我們再安裝一個更加強大的tensorflow-gpu版本，它可以調(diào)用conda的接口實現(xiàn)gpu運算的平臺,利用顯卡幫助我們運算程序，以提高后期學習中的程序處理速度,提高學習效率。

基礎知識信息了解

顯卡算力

????????在這個之前，我們首先要確保自己的電腦是英偉達顯卡，并且運算能力在3.5以上,大家可以根據(jù)下面的網(wǎng)址查看自己電腦顯卡的運算能力，然后還需要下載conda工具包和對應的gpu加速庫cuDNN。

直通車：CUDA GPUs - Compute Capability | NVIDIA Developer

????????后期安裝CUDA通過deviceQuery.exe也可以看到當前顯卡的算力。

CUDA兼容

????????這里CUDA12.1是支持的最高版本的CUDA，可以向下兼容，且可以安裝多個版本的CUDA,你可以通過更改環(huán)境變量來更改為你需要用到的CUDA版本。

Tensorflow gpu安裝

CUDA/cuDNN匹配和下載

查看Conda driver的版本

????????我們打開命令行窗口cmd，輸入nvidia-smi,這里顯示的是顯卡的版本信息，這里顯示的是conda driver的版本信息。

下載CUDA工具包

? ? ? ? 直通車：CUDA Toolkit Archive | NVIDIA Developer

????????我們?nèi)onda下載官網(wǎng)，下載CUDA工具包。根據(jù)剛剛我們查到的CUDA版本信息，此處我的CUDA版本為12.x，根據(jù)CUDA可以向下兼容的特性，我們可以對應下載CUDA11.4的工具包。

????????在這個界面，由于我的電腦是Windows11 64位，所以我選擇的是這些選項，大家要根據(jù)自己的電腦系統(tǒng)類型選擇合適的版本進行下載。

查看對應cuDNN版本

下面查找對應的cuDNN版本，可以在Tensorflows官網(wǎng)中查看tensorflow-gpu跟cuda?cudnn的版本對應信息.

在 Windows 環(huán)境中從源代碼構建 ?|? TensorFlow

下載cuDNN加速庫

????????接下來我們打開cuDNN下載地址：

直通車：https://developer.nvidia.com/rdp/cudnn-archive

????????由于我們CUDA是11的版本 對應的是cuDNN8的版本，這些版本的對應，小伙伴們一定要注意！現(xiàn)在我們打開cuDNN下載官網(wǎng)，在這里，根據(jù)剛剛查看到的cuda版本，選擇適當?shù)腸uDNN版本，我剛下載的是CUDA11.4的版本，也就是CUDA11.x的版本，所以我選擇的是cuDNN8.x的版本，這里我下載版本為8.2.1，然后選擇windows x86的選項進行下載。

? ? ? ? 注意：在這里點擊下載的時候會跳轉(zhuǎn)到注冊登錄頁面，由于在這里我已經(jīng)登錄，所以沒有跳轉(zhuǎn)，等待安裝包下載完成，我們就準備好了Tensorflow-gpu所需要的工具包，這就是已經(jīng)下載好的工具包。

????????如果各位小伙伴在這里遇到問題無法解決，可以在評論區(qū)進行求助。

CUDA/cuDNN安裝

CUDA安裝方法

????????接下來我們開始安裝CUDA，雙擊打開下載的安裝包，并等待進度條加載完畢。

????????點擊同意并繼續(xù)選擇自定義，然后點擊下一步。在這個界面顯示的是將要安裝的組件名稱、版本號和電腦中該組件的版本號，當前版本號為空，則說明電腦中沒有該組件。

????????我們取消NVIDIA GeForce Experience這一項，然后點擊下一步。這里的安裝路徑一般選擇默認就好，也可以更改，但是文件目錄一定要記清楚，后面配置環(huán)境的時候會用到。

然后點擊下一步，點擊next，等待安裝完成。

然后點擊下一步，這里顯示的是已經(jīng)安裝的所有組件的狀態(tài)，然后點擊關閉。

cuDNN加速庫安裝

????????下面開始安裝Gpu加速庫cuDNN，將文件解壓，解壓完成以后，我們打開會得到如下三個目錄。

? ? ? ? 然后我們打開剛剛安裝好的CUDA的根目錄，然后把codnn里邊并目錄下的所有文件復制到CUDA的bin目錄如下。

????????將include里邊的所有文件復制到CUDA的include文件下，lib文件夾也是如此。這樣我們便完成了CUDA和cuDNN的安裝。

配置CUDA/cuDNN環(huán)境變量

配置環(huán)境變量

????????下面開始設置系統(tǒng)環(huán)境變量，右鍵點擊此電腦，選擇屬性打開高級系統(tǒng)，設置環(huán)境變量，在系統(tǒng)變量里面找到path，點擊編輯。我們可以看到CUDA的兩個文件已經(jīng)存在，點擊新建瀏覽，找到CUDA目錄。

將其上移，與其他兩個環(huán)境變量一起，這樣就完成了環(huán)境變量的設置。

核驗是否安裝成功

????????可以通過nvcc -V命令查看是否配置CUDA成功

????????輸入nvidia-smi命令，返回GPU型號則安裝成功

? ? ? ? ?同時也可以通過在CUDA執(zhí)行bandwidthTest.exe和deviceQuery.exe和核驗，返回PASS則表明GPU安裝成功

Tensorflow-gpu安裝

????????接下來我們開始安裝tensorflow-gpu，安裝過程可以參考上篇文章tensnflow2.4的安裝，這里我就不再詳述。不同之處，就是我們創(chuàng)建并激活另一個獨立環(huán)境tensorflow-gpu選擇適當?shù)陌姹咎?#xff0c;小伙伴們可以根據(jù)自己的安裝環(huán)境選擇對應的版本安裝，第二步是安裝相關軟件，在第三步的時候安裝tensorflow-gpu，命令為pip?install?tensorflow-gpu==對應版本號安裝完成以后就完成了tensorflow-gpu的安裝。

直通車：人工智能發(fā)展歷程和工具搭建學習-CSDN博客

命令安裝

創(chuàng)建獨立環(huán)境并激活

conda create -n tensorflow-gpu python==3.8conda activate tensorflow-gpu

安裝相關軟件包

?
# conda install numpy matplotlib PIL scikit-learn pandas 于下行命令等價pip install numpy matplotlib Pillow scikit-learn pandas -i Simple Index

安裝Tensorflow-gpu

?
pip install tensorflow-gpu==2.6.0 -i Simple Index

報錯處理

????????ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts. matplotlib 3.7.5 requires numpy<2,>=1.20, but you have numpy 1.19.5 which is incompatible. pandas 2.0.3 requires numpy>=1.20.3; python_version < "3.10", but you have numpy 1.19.5 which is incompatible.

pip uninstall numpypip install numpy==1.19.5

????????TypeError: Descriptors cannot be created directly. If this call came from a _pb2.py file, your generated code is out of date and must be regenerated with protoc >= 3.19.0. If you cannot immediately regenerate your protos, some other possible workarounds are: 1. Downgrade the protobuf package to 3.20.x or lower. 2. Set PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python (but this will use pure-Python parsing and will be much slower).

pip uninstall protobuf pip install protobuf==3.20.0

校驗安裝結果

????????最后我們測試一下是否安裝成功，打開命令行窗口，激活我們剛才創(chuàng)建的獨立環(huán)境。輸入python，打開python交互模式，輸入import tensorflow as tf，輸入我們的測試語句tf.test.is_gpu_available()，它的輸出結果為true，顯示我們安裝成功。

python

import tensorflow as tftf.test.is_gpu_available()exit()