sw-gong/spiralnet_plus

Working Colab Notebook with updated dependencies!

nirajpandkar opened this issue · 0 comments

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 -

  • PyTorch 1.9 with CUDA 11.1
  • PyTorch Geometric 1.9 with CUDA 11.1 (Followed this handy script for the correct scatter and sparse CUDA versions)
  • OpenMesh 1.1.6
  • MPI-IS Mesh installed from source (instructions in the notebook)

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