系統需求

Linux系統，并已安裝git、gcc、g++、cmake?具

下載源碼

從github克隆Pytorch倉庫

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

如果已經克隆過，則使?以下命令更新

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

如?法更新?模塊，則需要?動克隆?模塊倉庫

（1）進?pytroch源碼?錄下的 third_party ?錄，查看未克隆成功的?模塊，如gloo
（2）刪除該?模塊?錄
（3）在https://github.com/pytorch/pytorch/tree/master/third_party處找到響應的?模塊，并點擊進 ?相應的倉庫，并使?git克隆到third_party?錄下，如

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

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

編譯環境

Android NDK

通過Android studio安裝相應的sdk：?具->SDK Manager->SDK Platforms

設置環境變量 ANDROID_HOME 為SDK安裝?錄（如上圖中的Android SDK Location ）
通過Android studio安裝相應的ndk：?具->SDK Manager->SDK Tools

設置環境變量 ANDROID_NDK 為NDK安裝?錄（?般在SDK安裝?錄下的ndk?錄下）

Gradle

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

JDK8

（1）從https://www.oracle.com/java/technologies/javase/javase-jdk8-downloads.html下載對應系統的JDK并安裝
（2）設置環境變量 JAVA_HOME 為Java安裝?錄
（3）在環境變量 PATH 中添加java安裝?錄下的bin ?錄

編譯Pytorch for Android

獲取模型運算符?件

（1）運?Reversi 項?中Predictor 類的save_script ?法，?成模型對應的運算符?件predictor.yaml
（2）將模型?件 predictor.yaml 復制到pytorch?錄下

編譯

（1）在pytorch?錄下運?以下命令?成aar?件

SELECTED_OP_LIST=predictor.yaml scripts/build_pytorch_android.sh x86

（2）命令參數x86 代表構建的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項?的app?件夾下新建libs ?錄，并復制pytorch_android-release.aar 到此?件夾下
（2）在項?的build.gradle配置?件夾中添加以下內容：

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

總結

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

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。