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 :)