TensorFlow 安裝環(huán)境搭建
目錄
安裝Anaconda
安裝CUDA
安裝cuDNN
配置環(huán)境變量
安裝TensorFlow

1. ?安裝Anaconda
Anaconda 下載:https://www.anaconda.com/products/distribution/start-coding-immediately
安裝到D盤(pán),節(jié)省空間

3.繼續(xù)勾選下面的選項(xiàng)

4. 安裝需要一些時(shí)間

5. 安裝成功后,按住鍵盤(pán)上的win+R,打開(kāi)運(yùn)行,輸入cmd
回車(chē)。打開(kāi)命令行窗口,輸入:conda list
,檢查是否成功安裝


2. 安裝cuda
- 檢查是否是`NVIDIA`顯卡
按住鍵盤(pán)上的ctrl+Shift+Esc,打開(kāi)任務(wù)管理器,選擇性能一欄,查看GPU。

2. 我的顯示有一塊NVIDIA顯卡,可以安裝CUDA
-下載CUDA
下載CUDA,依照我的電腦版本,選擇了win10,本地

2. 安裝CUDA,我選擇臨時(shí)解壓在D盤(pán)

3. 安裝選項(xiàng):




????????????4.選擇安裝位置

????5.????安裝完成后,打開(kāi)安裝文件夾D:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.1\bin
下的==nvcc.exe==,有該文件存在安裝基本

6. 打開(kāi)D:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.1\extras\CUPTI\lib64
,確認(rèn)是否存在==cupti64_2023.1.0.dll==動(dòng)態(tài)庫(kù)

檢測(cè)安裝是否成功
下載cuDNN,選擇與
CUDA
相匹配的版本,我這里安裝的CUDA v12.0,故選擇cuDNN v8.8.1 (March 8th, 2023), for CUDA 12.x
3. 安裝cuDNN


????2.??解壓壓縮包,并將解壓后的文件夾改名為cudnn
,并復(fù)制到D:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.1
文件夾下。


????3.確認(rèn)D:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.1\cudnn\bin
文件夾下是否有cudnn64_8.dll,這關(guān)系到tensorflow_GPU版能否安裝成功。

4.配置環(huán)境變量
1、打開(kāi)環(huán)境變量設(shè)置,如圖所示。

????????2、找到path,確定兩項(xiàng)是否存在

????3、添加其余兩項(xiàng),并將這四項(xiàng)置頂,否則會(huì)影響搜索,切四項(xiàng)缺一不可

????4、測(cè)試CUDA
,打開(kāi)命令行窗口,輸入nvcc -V
,顯示版本號(hào),則說(shuō)明安裝成功

打開(kāi)Anaconda Prompt窗口
5. 安裝TensorFlow

2. 安裝GPU版本的tensorflow,在命令行窗口輸入conda install tensorflow
因?yàn)榕f的tensorflow-GPU已經(jīng)被刪除了,目前只有tensorflow這個(gè)版本,也可以使用GPU計(jì)算,參考此連接https://pypi.org/project/tensorflow-gpu/
3.測(cè)試tensorflow是否安裝成功

