Android에서 Vulkan을 사용하고 싶습니다.

Module을 Load할 때
mModule = Module.load(assetFilePath(context, “vulkan.pt”), Device.VULKAN)

이렇게 하기위해

1.7 버전을 안드로이드 스튜디오에서
implementation ‘org.pytorch:pytorch_android:1.7.0’
implementation ‘org.pytorch:pytorch_android_torchvision:1.7.0’
했습니다.

outputTuple = mModule!!.forward(IValue.from(inputTensor)).toTuple()
forward하는 과정에서 에러가 발생합니다.

에러메시지는
com.facebook.jni.CppException: PyTorch is not linked with support for vulkan devices
Exception raised from getDeviceGuardImpl at …/c10/core/impl/DeviceGuardImplInterface.h:216 (most recent call first):

이거입니다.

이유가 무엇일까요?

1.7 버전은 예전 버전으로 보입니다.
1.9.0 버전이나 최근에 릴리즈된 1.10.0 버전으로 해보시면 어떨까요?

mModule = Module.load(assetFilePath(context, “vulkan.pt”), Device.VULKAN)

1.9버전은 위와 같은 load를 할때 device. vulkan을 설정할수가 없습니다.

1.9.0 이나 1.10.0 버전에 매개변수가 이전과 다르게 함수가 있어서 해봤으나 오히려 에러메시지도 제대로 표시하지도 못하고 프로그램이 죽는 현상이 있네요… 참 어렵네요