移植TensorFlow到Windows平台


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


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM