Issue with Importing madrona_gpudrive in Docker Setup

[niuyuchenf]

Hi,

I encountered an issue while following the installation instructions provided in the repository for setting up GPUDrive via Docker. After building the Docker container and installing everything as instructed, I ran the following Python command to check if the installation was successful:

import madrona_gpudrive

However, it results in the error: ModuleNotFoundError: No module named 'madrona_gpudrive'.

Interestingly, when I run the following:

import gpudrive

It works without any issues, and there is no error message.

Has anyone else encountered this problem? Is the madrona_gpudrive module perhaps missing or not properly installed in the Docker image? Could someone point me to the right steps to resolve this, or is there something I missed during installation?

Thanks in advance for any help!

[daphne-cornelisse]

Hi @niuyuchenf, thanks for your interest! I'm not familiar with this particular issue. @m-naumann do you have an idea?

[qizhantam]

Hi @niuyuchenf, I may have a solution. Try activating the virtual env first. In the docker (/gpudrive) directory: source .venv/bin/activate