Giter VIP home page Giter VIP logo

Comments (3)

zhan-xu avatar zhan-xu commented on September 16, 2024

could you please provide the version of cuda-toolkit you are using, as well as the versions of the installed pytorch, torch-scatter, torch-sparse, torch-cluster, torch-spline-conv, torch-geometric?

from rignet.

shshwdr avatar shshwdr commented on September 16, 2024

this are the versions I got:
CUDA 11.1
cuDNN v8.0.4
Pytorch: 1.7.0
Torch-scatter-2.0.5
Torch-sparse-0.6.8
Torch-cluster-1.5.8
Torch-spline-conv-1.2.0
torch-geometric-1.6.1

These are the commands I used:
pip install numpy scipy matplotlib tensorboard open3d==0.9.0 opencv-python
pip install Rtree‑0.9.4‑cp37‑cp37m‑win_amd64.whl
pip install trimesh[easy]
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
pip install torch-scatter==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-sparse==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-cluster==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-spline-conv==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-geometric

T

from rignet.

zhan-xu avatar zhan-xu commented on September 16, 2024

Thanks for sharing the details. It looks like you didn't install the correct version of torch-* corresponding to your cuda.

You are using cuda 11, so you will need to install torch-* for cuda110. This means the command should be modified into

pip install torch-scatter==latest+cu110 -f https://pytorch-geometric.com/whl/torch-1.7.0.html
pip install torch-sparse==latest+cu110 -f https://pytorch-geometric.com/whl/torch-1.7.0.html
pip install torch-cluster==latest+cu110 -f https://pytorch-geometric.com/whl/torch-1.7.0.html
pip install torch-spline-conv==latest+cu110 -f https://pytorch-geometric.com/whl/torch-1.7.0.html

Please also see https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html for installation instructions.

from rignet.

Related Issues (20)

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.