Closed zrxbeijing closed 4 years ago
Haven’t tried it yet. I probably would need to include the device as an argument
This project is just awesome! thanks, @dmmiller612, but it would be great to squeeze it to GPU, CPU takes forever to run.
So sorry this took so long. GPU support has just been added. I will be uploading the code to pypi in the next few hours. The current colab in the repo has an example with gpu support (If you have a cuda available, you shouldn't need to do anything on your part with the most recent code change).
So sorry this took so long. GPU support has just been added. I will be uploading the code to pypi in the next few hours. The current colab in the repo has an example with gpu support (If you have a cuda available, you shouldn't need to do anything on your part with the most recent code change).
Wow! That‘s awesome! Can’t wait to try it out!
Hello, I like your project very much. One question: how can we run the code with gpu support?