getkeops / keops

KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows
https://www.kernel-operations.io
MIT License
1.04k stars 64 forks source link

Add tip for adding nvcc to a working torch w/ conda #206

Open HastingsGreer opened 2 years ago

HastingsGreer commented 2 years ago

Hi! This is an approach that I found is reliably able to add pykeops to a working torch / conda installation, and which doesn't risk messing up the system-wide driver situation, especially on shared machines. Would you be interested in including it in the official documentation? Thanks for considering my pull request.

HastingsGreer commented 2 years ago

Inspiration from https://github.com/pyg-team/pytorch_geometric/issues/392#issuecomment-503335625