在windows上安裝tensorflow-gpu，安裝不上，一直提示
setuptools.extern.packaging.requirements.InvalidRequirement: Expected end or semicolon (after name and no valid version specifier)
python_version>"3.7",
只安裝了tensorflow=2.1.3,這里是成功的
但tf默認(rèn)是沒開啟GPU的,可以按win+r進(jìn)入cmd,啟動你的虛擬環(huán)境，輸入python用以下命令檢測：

'''
import tensorflow as tf
tf.test.is_gpu_available()
'''

如果最后顯示false,就是沒打開。
如果打開了，就是tensorflow可以使用GPU
好，按順序說下如何在windows下安裝tensorflow-gpu

硬件環(huán)境準(zhǔn)備。

1,劃重點:只能用cuda tool kit11.1
在查了官方說明后，在windows10/11上要安裝tensorflow-gpu只能用cuda tool kit11.1,更高版本不再支持，如果你一定要裝更高版本的只能在wsl2安裝。
cuda tool kit11.1下載
下載，解包安裝，點擊按按默認(rèn)安裝。
下載時需要注冊nvidia帳號，注冊很麻煩，但安全性很高。
安裝完成應(yīng)該能看見環(huán)境變量時增加如圖所示，

nvcc_test.png

(查找環(huán)境變量方法，win+i,進(jìn)入系統(tǒng)設(shè)置，到底部系統(tǒng)信息，高級系統(tǒng)設(shè)置,環(huán)境變量)，

安裝成功后，win+R進(jìn)入cmd,輸入nvidia-smi應(yīng)該看到下述版本號：12.2

nvidia-smi.png

我的版本是12.2就是因為前面先裝了12.2，后來卸裁了，重裝了11.1版本，但這里還是顯示12.2,這里重點是：如果輸入nvidia-smi看得到cuda took kit版本號就可以認(rèn)為成功了。如果沒看到重啟下試試。

2,安裝cudnn,按照官方說明，cuda tool kit 11.1對應(yīng)的版本cudnn版本為8.0,在下面這個鏈接
cudnn安裝

找到你的版本對應(yīng)的cudnn版本，我的是8.0.5，下載后是個zip文件，解壓zip文件，得到三個目錄，把這三個目錄拷貝到：
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1
這里是默認(rèn)安裝，并且版本是v11.1
按你自己情況改。
改完之后，在系統(tǒng)變量里，PATH增加如下：
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11\lib
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11\include

path_cuda.png

做完這些重啟。
然后win+R進(jìn)入cmd
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\extras\demo_suite

找到deviceQuery.exe,運行應(yīng)出現(xiàn)result=Pass頁面

devicequery.png

這里可以看到cuda toolKit版本是12.2，但運行庫是11.1,所以可以運行tensorflow-gpu的。

找到bandwidthTest，運行它

bandwidth.png

也應(yīng)該看到result=pass。
OK,這樣可以認(rèn)為你的開發(fā)環(huán)境的硬件驅(qū)動成功了。

軟件部分

現(xiàn)在安裝軟件部分，就簡單了，下載conda安裝成功，啟動虛擬環(huán)境。這里有一點python版本>3.7,小于3.11，GPT回答我3.11不支持，我沒試，但我安裝了3.8,然后安裝
'''
pip install tensorflow-gpu==2.1.0
'''
網(wǎng)絡(luò)環(huán)境有時不通，多試幾次
安裝成功后，按上面所說測試
'''
import tensorflow as tf
tf.test.is_gpu_available()
'''

tf_ture.png

至此成功，總結(jié)
1,windows只支持cuda tool kit 11.1
2,cuda tool kit對應(yīng)版本是8.0.5
3,tensorflow安裝版本小于2.1.0，安裝時是
pip install tensorflow-gpu,如果不加GPU,默認(rèn)裝的是CPU版本，加了GPU,引用的還是tensorflow,這時啟用了GPU
注意這二點，其他是常規(guī)操作。

我蘇格，跨境電商數(shù)據(jù)提供商。關(guān)注:SEO,反焦慮,內(nèi)向，就是下樓前聽下鄰居有沒有也準(zhǔn)備下去