本文主要針對(duì)在windows10環(huán)境下的tensorflow配置問(wèn)題，在linux和mac等其他環(huán)境中的配置就不過(guò)多贅述(windows總是那個(gè)問(wèn)題最多的環(huán)境，建議使用linux 😃)。

本文中配置的環(huán)境為 python 3.8.5 tensorflow-gpu 2.4.1

1. 更新nvidia顯卡驅(qū)動(dòng)至最新

測(cè)試用顯卡為 RTX 2070，如果不確定自己的顯卡是否適配，可以登錄nvidia官網(wǎng)進(jìn)行查詢。

2. 下載cuda 11.0

[cuda 11.0 下載地址]
注意適配tensorflow 2.4 版本的cuda 一定是11.0，而不是11.1，11.2或者其他版本，下載其他版本的cuda會(huì)導(dǎo)致庫(kù)缺失的問(wèn)題。
cuda的默認(rèn)安裝路徑為 C:\Program Files\NVIDIA GPU Computing Toolkit

3. 下載cudnn 8.0.4用于適配TF 2.4 以及cuda 11.0

[cudnn 8.0.4 下載地址]
下載完畢cudnn后解壓，分別將子文件夾 bin，include，lib/x64中的文件拷貝至cuda/v11.0 對(duì)應(yīng)的同名文件夾中。

4. 使用anaconda為tensorflow-gpu 配置一個(gè)python環(huán)境

# 創(chuàng)建一個(gè)名稱為 tf 版本為3.8.5的anaconda python環(huán)境 conda create -c conda-forge python=3.8.5 -n tf conda activate tf

安裝版本為2.4.1的tensorflow-gpu

pip install tensorflow-gpu==2.4.1

5. 常見(jiàn)問(wèn)題

1.

Could not load dynamic library 'cudart64_110.dll'; dlerror: cudart64_110.dll not found

這個(gè)問(wèn)題一般是由于cuda版本與tensorflow gpu不兼容導(dǎo)致的，如果沒(méi)有使用tensorflow2.4而是使用了舊版的tensorflow。可以參考下表重新下載對(duì)應(yīng)的cuda配件。多個(gè)cuda版本是不會(huì)互相影響的，因此不用擔(dān)心cuda文件夾中存在多個(gè)不同的版本。
遇見(jiàn)這個(gè)報(bào)錯(cuò)一定不要僅僅只復(fù)制了這個(gè)dll配置文件到指定文件夾，這會(huì)導(dǎo)致隱形bug，tensorflow會(huì)在進(jìn)入feed數(shù)據(jù)階段時(shí)停止。

2. 測(cè)試環(huán)境

用于識(shí)別目前available的gpu的個(gè)數(shù)。

print("Num GPUs Available: ", len(tf.config.list_physical_devices('GPU')))

確認(rèn)所有的庫(kù)都可以成功打開(kāi)且可用gpu個(gè)數(shù)不為0即可宣告配置完成。

2021-03-15 15:12:17.673193: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library cudart64_110.dll 2021-03-15 15:12:17.691692: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library cublas64_11.dll 2021-03-15 15:12:17.698596: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library cublasLt64_11.dll 2021-03-15 15:12:17.711218: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library cufft64_10.dll 2021-03-15 15:12:17.720665: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library curand64_10.dll 2021-03-15 15:12:17.735681: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library cusolver64_10.dll 2021-03-15 15:12:17.746699: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library cusparse64_11.dll 2021-03-15 15:12:17.754405: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library cudnn64_8.dll 2021-03-15 15:12:17.759877: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0 Num GPUs Available: 1

其他的anaconda中配置環(huán)境的代碼可以參考下文 :
[Anaconda安裝tensorflow報(bào)錯(cuò)問(wèn)題解決方法]

總結(jié)

以上是生活随笔為你收集整理的tensorflow gpu windows配置步骤教学的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。