Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Docker image with a working environment #29

Open
MTDzi opened this issue Aug 29, 2022 · 3 comments
Open

Docker image with a working environment #29

MTDzi opened this issue Aug 29, 2022 · 3 comments

Comments

@MTDzi
Copy link

MTDzi commented Aug 29, 2022

First of all: awesome work, really appreciate it! :)

Second: I was trying to get it to work but I was facing issues installing the right versions of the dependencies, so I figured I'll create a docker image that might work not only for me.

If you guys are interested, I can create a PR with the Dockerfile that worked for me.

@Henry1iu
Copy link
Owner

Hi,

Thanks for your appreciation!

Installing the correct dependencies for torch_geometric is kind of tricky. Can you share the error message? I'll try my best to help you! :-)

Best Regards,
Jianbang

@MTDzi
Copy link
Author

MTDzi commented Aug 30, 2022

Hmmm... so I did run into a bunch of different errors, all of them were caused by some mismatch in the versions (and I tried the requirements.txt but maybe it also requires the proper CUDA version? IDK, either way it didn't work).

And I wanted to share the setup that worked for me personally, so that anyone in the future might benefit from it.

The Dockerfile that worked for me is essentially the following (it's a bit WIP, but the hardest part is there):

ARG FROM_IMAGE_NAME=nvcr.io/nvidia/pytorch:20.12-py3

FROM ${FROM_IMAGE_NAME}

RUN update-alternatives --install /usr/bin/python python $(which python3) 1

RUN pip3 install --upgrade pip && pip3 install \
    torch==1.9.1+cu111 \
    torchaudio \
    -f https://download.pytorch.org/whl/cu111/torch_stable.html

RUN pip3 install --upgrade pip && pip3 install \
    lmdb \
    torch-scatter==2.0.9 \
    -f https://pytorch-geometric.com/whl/torch-1.9.1+cu111.html
RUN pip3 install --upgrade pip && pip3 install \
    torch-sparse==0.6.12 \
    -f https://pytorch-geometric.com/whl/torch-1.9.1+cu111.html
RUN pip3 install --upgrade pip && pip3 install torch-geometric==1.7.2

RUN git clone https://github.com/argoai/argoverse-api && pip3 install argoverse-api/
RUN pip3 uninstall -y opencv-python && pip3 install opencv-python

So the question is: would you guys benefit from a PR with a Dockerfile (a more polished version than the one above)? Or do you simply don't work in a devcontainer and there's no need for that?

@Ceudan
Copy link

Ceudan commented Jul 4, 2023

I believe many of us users are also struggling with torch_geometric dependencies. We would definitely benefit from a Dockerfile.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants