怎么安裝tensorflow？

時間：2023-10-26 17:18:01 | 來源：網(wǎng)站運營

時間：2023-10-26 17:18:01 來源：網(wǎng)站運營

怎么安裝tensorflow？：

前情提要：

這幾天實驗室新配了一臺3090的工作站，說是工作站，其實是游戲主機（GPU實在是太難買了！）。所以跟隨主機過來的是Windows系統(tǒng)?？紤]到保修協(xié)議，一年內(nèi)不能刷成Linux系統(tǒng)，就只能琢磨一下怎么在Windows配置TensorFlow了。此文用于記錄在我嘗試配置成功后的每一步，也分享給有需要的大家。

為什么需要一個教程？

因為如果你按照官方pip一鍵安裝：

pip install tensorflow你一定會發(fā)現(xiàn)你的模型沒有在使用GPU。其實官方也說得很清楚，需要你Build from source。

為什么不用Conda命令來安裝?

Conda很好，按照這個命令，確實可以安裝出能使用GPU的版本

conda install -c anaconda tensorflow-gpu因為Conda目前只更新到2.6（我自己測試了一下，給我裝了2.1，我很迷惑），所以如果你對于版本要求沒有很嚴格，可以采取這個方案，請點贊然后不用閱讀后面的啦。

本文適合誰？

需要TensorFlow-GPU 2.6+的使用Windows系統(tǒng)的同學(xué)。需要一定的基礎(chǔ)知識，比如命令行的使用。

聲明：

本文主要根據(jù)這兩個信息源：

沒有墻且英語好的同學(xué)也可以直接去這兩個教程。

開始吧。

第一步：更新你的NVIDIA驅(qū)動

這一步因為我是GeForce RTX系列，所以我直接使用NVIDIA GeForce Experience軟件完成安裝。

使用別的系列的卡的同學(xué)可以去 Download the latest official NVIDIA drivers 下載驅(qū)動然后安裝。

第二步：安裝Visual Studio

這一步驟的意義根據(jù)Jeff Heaton教授所說是為了讓我們的系統(tǒng)有一個可用的C++編譯器。注意是Visual Studio，不是VS Code。

在安裝的時候不用選任何的Workload，Jeff建議選擇如下Individual Components:

點擊Install然后會有一個這個框框：

因為我們不需要Workload，所以直接點擊Continue就可以開始安裝了。

16MB每秒的速度，還可以。

安裝完成后找到安裝目錄，我的在這里：C:/Program Files/Microsoft Visual Studio

找到這個文件：C:/Program Files/Microsoft Visual Studio/2022/Community/VC/Auxiliary/Build/vcvars64

打開CMD，輸入：

cd C:/Program Files/Microsoft Visual Studio/2022/Community/VC/Auxiliary/Build/然后輸入：

vcvars64.bat你應(yīng)當(dāng)看到如下信息：

說明咱們的C++編譯器就設(shè)定好了。

第三步：安裝CUDA11.2

雖然已經(jīng)有11.6，但是目前官方還是推薦使用CUDA11.2+CuDNN8.1的配置。

去這里找11.2的安裝包：

CUDA Toolkit Archive下載然后安裝。

我裝的是11.2.2，這個安裝包很無腦，選擇Express安裝然后按照默認一路點下去就好了。

第四步：安裝CuDNN8.1

這一步就很麻煩了，首先你要在這里下載CuDNN 8.1.1的 .zip文件：

NVIDIA會讓你注冊一個開發(fā)者賬號然后才能下載。（據(jù)我所知墻內(nèi)的同學(xué)可以使用XX鏡像？）

點擊我下劃線的那個，你會下載一個壓縮文件。重點來了，下面是我配置成功的技巧，可能還可以化簡，但是我就這么配置了：

1. 你需要在你的C盤建立一個叫做tools的文件夾。
2. 解壓剛剛下載的壓縮文件到tools里面。
3. 打開系統(tǒng)的路徑管理器，長這樣：

4. 打開Environment Variables，在System Variables找到Path，然后點擊Edit。

5.將以下內(nèi)容加入進你的Path中：

C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.2/binC:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.2/extras/CUPTI/lib64C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.2/extras/CUPTI/includeC:/tools/cudnn-11.2-windows-x64-v8.1.1.33/cuda/binC:/tools/cudnn-11.2-windows-x64-v8.1.1.33/cuda/include注意檢查一下，大部分同學(xué)應(yīng)該在path里有第一條了。同時這里的路徑是我自己的（一般也是默認安裝后的），如果不一致需要自己更改成你們的。

第五步：確認PATH

這一步很重要，打開你們的CMD，按照這個方式挨個測試你們的系統(tǒng)能訪問上面五個路徑：

dir "PATH"把“”里面的PATH換成每個path，比如這樣：

如果有這樣的信息，說明路徑是可以訪問的，沒有問題。

第六步：復(fù)制dll文件

有些人說這個沒必要，但是我建議還是這么做，算是雙保險。

1. 把C:/tools/cudnn-11.2-windows-x64-v8.1.1.33/cuda/bin里的全部.dll文件復(fù)制
2. 粘貼到C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.2/bin

第七步：安裝TensorFlow 2.8

終于到這一步了。

建議用Conda建立一個新的環(huán)境，我的叫做tf2:

conda create --name tf2conda activate tf2然后

pip install tensorflow等待安裝完成后，測試一下：

成功！

結(jié)尾的一些話

這是我第一次在本站寫技術(shù)類長文，還有很多不妥、講的不細節(jié)的地方，希望大家多多包涵！