2023.6 結(jié)論
- WSL+ubuntu（或者純ubuntu）可以解決大部分問題逢享，比如ubuntu20.04默認(rèn)的python3.8下可以支持：
Tensorflow_federated 0.33+TF2.9+cuda11.8
不玩TFF，可以上TF2.11吴藻，cuda11.8也支持瞒爬。
pytorch2當(dāng)前只支持到cuda11.8，所以不需要搞得太高版本沟堡。
2.在windows下侧但，也是建議cuda11.8+cudnn8.8（這個高低一些低一些好像都行），同時可以支持pytorch2和TF2.10航罗，但TFF只能玩0.17版禀横。
| TF | Cuda | Cudnn | Keras |
|---|---|---|---|
| tensorflow2.10.x | cuda11.8.0 | cudnn8.8.0(for cuda11.x) | keras2.10.0 |
需要注意的是：
- Tensorflow 2.10是最后一個在windows上支持GPU的版本，更高版本的TF在windows上找不到GPU粥血，最好轉(zhuǎn)到ubuntu柏锄。
- 這個版本的CUDA在windows上可能出“Could not locate zlibwapi.dll“ 錯誤酿箭，解決方法：
- 下載 http://www.winimage.com/zLibDll/zlib123dllx64.zip
（這個連接來自官方文檔[Installation Guide :: NVIDIA Deep Learning cuDNN Documentation）(https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html ）
2，里面的zlibwapi.lib文件拷貝到cuda安裝目錄的lib文件夾趾娃，zlibwapi.dll文件拷貝到cuda安裝目錄的bin文件夾即可缭嫡，壓縮包內(nèi)別的文件都不用管。
如果不是歷史問題抬闷，下面的東西不太用看了妇蛀，不太需要考慮多版本cuda。
-----------------------下面是一些老皇歷了--------------------
多版本tensorflow+cuda+cuDNN共存環(huán)境
python版本
- python 3.8.10可以支持到TF 2.6（但在wsl上測試饶氏，tf2.11也能跑）讥耗，2.10以上推薦使用python3.9以上，具體懶得細(xì)致整理了疹启。
- TF1建議用pyhton3.6.x，用python3.8.x 蔼卡，pip install tensorflow==1.x.x 提示找不到對應(yīng)版本喊崖。
之前測試過的的TF+Keras+CUDA版本
| TF | Cuda | Cudnn | Keras |
|---|---|---|---|
| tensorflow1.14 | cuda10.0.x | cudnn7.4.2（for cuda10） | keras2.2.5 |
| tensorflow2.3.4 | cuda10.1.x | cudnn7.6.5（for cuda10.1） | keras2.4.3 |
| tensorflow2.6.3 | cuda11.2.x | cudnn8.1.1 | keras2.6.0 |
windows玩聯(lián)邦學(xué)習(xí)
| TTF | TF | 平臺 |
|---|---|---|
| -tensorflow-federated 0.17 | tensorflow2.3.4 | Windows |
tff目前最新版本是0.48，但里面涉及到一些包只支持linux雇逞，windows下還沒找到好的安裝方法荤懂。
顯卡
- 親測3060不能很好的支持cuda10.1，會卡在莫名其妙的地方塘砸。
- 嘗試跑了一個非圖像的項(xiàng)目节仿，3060比2060只快了7%。不過可能是風(fēng)扇更大掉蔬，溫度保持在50多度廊宪，2060要80度了。
- 顯卡驅(qū)動女轿，據(jù)說 studio版本比game ready的要好箭启，也有人說差不多。
cuda+cudnn多版本切換
- 打開windows的系統(tǒng)環(huán)境變量蛉迹，看到CUDA_PATH這里傅寡，根據(jù)想用的cuda版本，選擇合適的路徑北救。
image.png -
然后在點(diǎn)開path環(huán)境變量：
image.png - 最上面四個cuda相關(guān)的路徑荐操，兩個關(guān)于10.1，兩個關(guān)于10.0珍策，用誰就把相應(yīng)的版本上移托启。（這個步驟有點(diǎn)玄學(xué)，可能沒用膛壹，但保險）驾中。
- 最后重啟pycharm唉堪，怕不保險就重啟電腦。
- 此外肩民，還有人說要修改NVCUDASAMPLES_ROOT這個環(huán)境變量唠亚，但目前感覺不用
