系統(tǒng)需求

Linux系統(tǒng)，并已安裝git、gcc、g++、cmake?具

下載源碼

從github克隆Pytorch倉(cāng)庫(kù)

git clone --recursive https://github.com/pytorch/pytorch

如果已經(jīng)克隆過(guò)，則使?以下命令更新

git submodule sync git submodule update --init --recursive

如?法更新?模塊，則需要?動(dòng)克隆?模塊倉(cāng)庫(kù)

（1）進(jìn)?pytroch源碼?錄下的 third_party ?錄，查看未克隆成功的?模塊，如gloo
（2）刪除該?模塊?錄
（3）在https://github.com/pytorch/pytorch/tree/master/third_party處找到響應(yīng)的?模塊，并點(diǎn)擊進(jìn) ?相應(yīng)的倉(cāng)庫(kù)，并使?git克隆到third_party?錄下，如

git clone https://github.com/facebookincubator/gloo.git

（4）重復(fù)以上操作直到third_party?錄下所有?模塊克隆成功

編譯環(huán)境

Android NDK

通過(guò)Android studio安裝相應(yīng)的sdk：?具->SDK Manager->SDK Platforms

設(shè)置環(huán)境變量 ANDROID_HOME 為SDK安裝?錄（如上圖中的Android SDK Location ）
通過(guò)Android studio安裝相應(yīng)的ndk：?具->SDK Manager->SDK Tools

設(shè)置環(huán)境變量 ANDROID_NDK 為NDK安裝?錄（?般在SDK安裝?錄下的ndk?錄下）

Gradle

（1）從https://gradle.org/releases/下載相應(yīng)版本（如v6.8.3）的gradle壓縮包并解壓
（2）設(shè)置環(huán)境變量 GRADLE_HOME 為解壓后的gradle?錄
（3）在環(huán)境變量 PATH 中添加gradle?錄下的bin ?錄

JDK8

（1）從https://www.oracle.com/java/technologies/javase/javase-jdk8-downloads.html下載對(duì)應(yīng)系統(tǒng)的JDK并安裝
（2）設(shè)置環(huán)境變量 JAVA_HOME 為Java安裝?錄
（3）在環(huán)境變量 PATH 中添加java安裝?錄下的bin ?錄

編譯Pytorch for Android

獲取模型運(yùn)算符?件

（1）運(yùn)?Reversi 項(xiàng)?中Predictor 類(lèi)的save_script ?法，?成模型對(duì)應(yīng)的運(yùn)算符?件predictor.yaml
（2）將模型?件 predictor.yaml 復(fù)制到pytorch?錄下

編譯

（1）在pytorch?錄下運(yùn)?以下命令?成aar?件

SELECTED_OP_LIST=predictor.yaml scripts/build_pytorch_android.sh x86

（2）命令參數(shù)x86 代表構(gòu)建的Android ABI，可選的ABI有：x86、x86_64、armeabi-v7a、arm64- v8a
（3）若編譯成功，將在pytorch/android/pytorch_android/build/outputs/arr下找到pytorch_android- release.arr ?件

使?編譯后的aar?件

（1）在Android項(xiàng)?的app?件夾下新建libs ?錄，并復(fù)制pytorch_android-release.aar 到此?件夾下
（2）在項(xiàng)?的build.gradle配置?件夾中添加以下內(nèi)容：

repositories { flatDir { dirs 'libs' }jcenter() }

在build.gradle配置?件中的dependencies下添加aar?件，并注釋掉之前使?的pytorch，如：

dependencies { implementation 'androidx.appcompat:appcompat:1.1.0' api(name: 'pytorch_android-release', ext: 'aar') // implementation 'org.pytorch:pytorch_android:1.8.0' implementation 'junit:junit:4.12' implementation 'com.alibaba:fastjson:1.1.70.android' }

參考鏈接

https://pytorch.org/mobile/android/
https://zhuanlan.zhihu.com/p/299736532

總結(jié)

以上是生活随笔為你收集整理的黑白棋游戏水平(2)--pytorch剪裁的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。