Is there a gpu version of WHL? I'm running on rk3399 (arm64)

lhelontra / tensorflow-on-arm

TensorFlow for Arm

MIT License

1.05k stars 274 forks source link

Closed gzchenjiajun closed 4 years ago

gzchenjiajun commented 4 years ago

Is there a gpu version of WHL? I'm running on rk3399 (arm64)

lhelontra commented 4 years ago

No, if you wants uses gpu, i recommended uses tvm. Tvm imports models from tensorflow and another frameworks, provide good opencl support.