sw-gong / spiralnet_plus

SpiralNet++: A Fast and Highly Efficient Mesh Convolution Operator (ICCV-W 2019)
https://arxiv.org/abs/1911.05856
MIT License
123 stars 23 forks source link

Working Colab Notebook with updated dependencies! #9

Open nirajpandkar opened 3 years ago

nirajpandkar commented 3 years ago

Hello,

This is more of an observation than an issue. I spent a considerable time setting up the environment and getting everything to work correctly (especially the CUDA versions of Pytorch and Pytorch-Scatter and Pytorch-Sparse). I have everything working in a Google Colab Notebook. But I would like to make a Docker Image file with all the specifications for easier onboarding. (And to get it working on my university servers) Although the 13 GB RAM is insufficient for running python -m reconstruction.main to completion, at least we know it's working.

Also, if you are convinced about the dependencies, I could update the list of dependencies in the README. Let me know if you want me to put in a pull request :)

I have installed the environment on Google Colab successfully with the following specifications -

I'll update here once I have a working Dockerfile with the above environment. Cheers :)