多分こんな感じで導入出来るはず
でもうちの環境はCUDA対応してないから動作確認は出来てない
CPUでも動かせるかと思ったけど動かなかった

https://pastebin.com/LxcKVMmJ