Theseus installation on Google Colab

[saxenam06]

Hello, Do you have any plans to extend the Theseus installation in Google Colab. The main issue are follows which restricts usage of Theseus on colab.

1. Colab doesn't support docker yet so it's not possible to build the wheels with the build script.
2. Colab cuda version is 11.2 which is not supported by the build script.
3. I tried to change cuda version to 10.2 so as to install Theseus with pip. Though the installation complete successfully But I get error while Importing Theseus: cannot import name '__C' from functorch. .

I would highly appreciate your thoughts on the same. Many thanks!

[mhmukadam]

Hi, you can easily install Theseus using the commands below to run in colab:

!apt install libsuitesparse-dev
!pip install theseus-ai

import torch
import theseus as th

th.rand_se2(2) # to test

The cuda version issue happened recently as torch 1.13 stoped supporting cuda 10.2. We are fixing this in #345 and should be merged soon.

[luisenp]

Hi @saxenam06. This error is related to the CUDA version issue @mhmukadam mentioned above, which also affects our pypi wheels. This issue should be solved when we release 0.1.3, which will happen soon (maybe even today).

Were you able to change CUDA version in Colab? We are still planning to keep 10.2 as the default for now, but curious to know if this would be an issue.

[saxenam06]

Hi @luisenp, Many Thanks for your response. Yes, I was able to change the CUDA version in Colab to 10.2. This is what I get in Colab with !nvcc --version:

But after this when I follow the below steps as @mhmukadam mentioned above, I get the error below that Failed building wheels for theseus-ai. So basically the below steps works only with a fresh Colab runtime without any changes done in the CUDA version. Thanks again for your time into this issue!

!apt install libsuitesparse-dev
!pip install theseus-ai

[luisenp]

Yes, this is expected. Should be fixed in 0.1.3.

[saxenam06]

Thanks a lot for your fix. It works now at my end. With CUDA version changed to 10.2 in COLAB and then installing torch, libsuitesparse and Theseus in a Conda Python 3.9 environment, I am able to run my scripts without solver/torch/Theseus build related issues mentioned before. Thanks a lot for your continuous support.

[luisenp]

Glad to hear that it works, and happy to help!