2015年11月,Google宣布開源旗下機器學習工具TensorFlow,引發業界熱潮。TensorFlow原生支持*unix系和安卓平台,但並不提供對Windows平台的支持。如果想在Windows上運行TensorFlow,官方提供了Docker/Linux虛擬機的繞行方案。
2016年4月,出於工作的安排,我承擔了將TensorFlow(當時的版本是0.8)移植到Windows平台的任務。TensorFlow項目的代碼包括C++內核和Python API兩部分,前者用於科學計算,后者簡化DAG構建。在此之前,我一直混跡在JAVA陣營,對於C++和Python的知識局限在大學課本的水平。因此,這個移植任務對我而言還是難度很大的,跨平台的語法、編譯、鏈接、測試、打包等等一切都要靠自己慢慢摸索。
歷時兩個月,在7月初我獨自一人完成了TensorFlow初版的移植,運行情況良好。由於性能有限以及團隊缺少資源維護代碼的時效性等原因,我的這一產出並沒有最終送入生產環境。除了提供給另一團隊進行TensorFlow的擴展開發外,基本被擱置了——目前團隊集中精力在另一套TensorFlow跨平台的技術方案上。
為了不讓我的這份工作浪費,我將自己的產出開源到GitHub上,對機器學習或者跨平台移植感興趣的同學可以自由的clone到本地並進行編譯、打包、安裝和使用。
項目地址:https://github.com/shishaochen/TensorFlow-0.8-Win
