lltcggie / waifu2x-caffe

waifu2xのCaffe版
MIT License
8.04k stars 839 forks source link

Any chance to support tensor cores? #124

Open fnscar opened 5 years ago

lltcggie commented 5 years ago

お金に余裕ができてTensorコアを搭載したグラボを購入出来たら検証予定です。

ChrisG683 commented 5 years ago

Hope to hear more on this! The potential of Tensor cores on the RTX 2XXX series seems very intriguing!

TheDuriel commented 4 years ago

In case someone would want to integrate this. It seems relatively straight forward on CuDNN 7.6.5 and above.

Relevant documentation overview: https://docs.nvidia.com/deeplearning/sdk/cudnn-developer-guide/index.html#tensor-ops-conv-functions