pretty code

2020年4月21日 星期二

TensorFlow 2.1 新變更

最近有在 Windows 透過 pip 安裝 TensorFlow 的人應該都會發現,安裝時名稱沒有帶 '-gpu',但安裝的貌似是 GPU 版,故執行時都會跳出下面的警告:

W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'cudart64_100.dll'; dlerror: cudart64_100.dll not found

在警告下面,TensorFlow 也很貼心的提醒可以忽略它:

I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.

雖然猜測是 GPU 版,但一直不知道為什麼?直到看到 stackoverflow 的某篇文章才了解其來龍去脈。

簡單來說,從 TensorFlow 2.1.0 發佈後,pip 在 Linux 和 Windows 上預設都是 GPU 版,而這個版本有沒有 NVIDIA GPU 都可以執行,而原本的 'tensorflow-gpu' 名稱仍然存在,如果想要減少安裝空間,也可以單獨安裝 CPU 版,名稱則是 'tensorflow-cpu'。

即使安裝的是舊版本(1.x),看起來預設也是 GPU 版,至少我在 Win10 安裝的 1.15.0 是如此。

沒有留言: