GithubHelp home page GithubHelp logo

Forces CUDA Device 0 about libmolgrid HOT 4 CLOSED

gnina avatar gnina commented on August 11, 2024
Forces CUDA Device 0

from libmolgrid.

Comments (4)

dkoes avatar dkoes commented on August 11, 2024

With 0.5.1 more informative error messages are generated if a torch tensor isn't on the current GPU device and helper functions (molgrid.set_gpu_device and molgrid.get_gpu_device) are now available for changing the current GPU.

I think it is best practice to make the programmer explicitly decide what GPU things are on, so calling molgrid.set_gpu_device is going to be the solution rather than implicitly changing GPU devices to match the tensor.

from libmolgrid.

RMeli avatar RMeli commented on August 11, 2024

Are molgrid.set_gpu_device and molgrid.get_gpu_device documented anywhere? I think it would be beneficial to add molgrid.set_gpu_device to the tutorial (and the documentation if not already there, I was unable to find them).

from libmolgrid.

dkoes avatar dkoes commented on August 11, 2024

I needed to add these methods to the sphinx rst file for them to show up in the documentation. I've pushed the change. I'd gladly accept a pull request for a more comprehensive getting started tutorial (it's on the todo list, but happy for someone else to do it)...

from libmolgrid.

RMeli avatar RMeli commented on August 11, 2024

Thanks, it will be very useful to have these functions documented.

I almost finished a PyTorch-based implementation of the GNINA scoring function (to be extended for flexible docking) so maybe a subset of that could be used as a starting point for a more comprehensive tutorial? (However I used PyTorch-Ignite quite heavily, which might obfuscate molgrid concepts...)

I've been wanting to put my own notes about GNINA and molgrid into some sort of "user guide" but I haven't yet had time to think more concretely about it either.

from libmolgrid.

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.