Windows下編譯TensorFlow2.0的C庫(kù)

1、安裝python

Python2和Python3都可以,我使用的是Python2.7。

2、安裝MSYS2

在Windows系統(tǒng)下搭建一個(gè)類(lèi)Linux環(huán)境,下載地址:http://www.msys2.org/,下載如圖1版本,并安裝。

圖1

將目錄C:\msys64\usr\bin 加入系統(tǒng)環(huán)境變量path;

打開(kāi)cmd.exe,輸入命令:

pacman -S patch,unzip,grep,git

注:如果pacman命令一直安裝不上,請(qǐng)單獨(dú)一個(gè)一個(gè)安裝。

3、安裝VS2015

VS至少要2015及其以后的版本。

4、安裝bazel

到github上下載Bazel,https://github.com/bazelbuild/bazel/releases,當(dāng)前準(zhǔn)備編譯TensorFlow2.0,下載的是bazel-0.26.1-windows-x86_64.exe。

將下載好的bazel-0.26.1-windows-x86_64.exe復(fù)制到C:\msys64下,更名為bazel.exe。

將C:\msys64加入系統(tǒng)環(huán)境變量。

配置bazel:新建環(huán)境變量:BAZEL_SH,BAZEL_VC,BAZEL_VS,如圖2。

圖2

5、下載TensorFlow2.0

下載地址:https://github.com/tensorflow/tensorflow,Branch中選擇r2.0,下載源碼,如圖3。

圖3

6、安裝CUDA10和CUDNN支持GPU

只編譯CPU版本可略過(guò)該步驟,TensorFlow2.0需要CUDA10的支持,CUDA的安裝就不詳說(shuō)了,需要注意的是需要安裝對(duì)應(yīng)的CUDNN版本,此次選擇cudnn-10.0-windows10-x64-v7.6.3.30.zip。

將解壓后的cudnn對(duì)應(yīng)的bin、include、lib目錄下的文件拷貝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0對(duì)應(yīng)的目錄。

7、配置編譯系統(tǒng)

進(jìn)入TensorFlow2.0的源碼目錄,打開(kāi)cmd.exe,執(zhí)行python ./configure.py,如圖4。

圖4

8、編譯

編譯生成CPU版本:

bazel build --config=opt --copt=-nvcc_options=disable-warnings //tensorflow/tools/lib_package:libtensorflow

編譯生成GPU版本:

bazel build --config=opt --config=cuda --copt=-nvcc_options=disable-warnings //tensorflow/tool/lib_package:libtensorflow

注:編譯的過(guò)程中會(huì)遇到編譯報(bào)錯(cuò),請(qǐng)找到相應(yīng)的源碼,改正后繼續(xù)編譯。編譯錯(cuò)誤的解決方案可參見(jiàn)bing搜索到的結(jié)果。

完成編譯后,就生成了可在Windows上使用的TensorFlow2.0的c庫(kù)。

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀(guān)點(diǎn),簡(jiǎn)書(shū)系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

友情鏈接更多精彩內(nèi)容