Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

Update CUDA search path to pick up `cudatoolkit` in Conda installs

See original GitHub issue

Based on discussion starting here, it appears cudatoolkit is not getting picked up. This could easily be fixed by adding sys.prefix to the CUDA search paths.

cc @leofang @gmarkall

Issue Analytics

State:
Created 3 years ago
Comments:9 (9 by maintainers)

Top GitHub Comments

1reaction

leofangcommented, Apr 8, 2020

For my own note:

Search strategy of PyTorch: https://github.com/pytorch/pytorch/blob/5d1205bf02296237f7ac556a6269283aa55e2a36/torch/utils/cpp_extension.py#L24-L51
Search strategy of cuDF (and rmm): https://github.com/rapidsai/cudf/blob/f5b7ed611fa8cc260ad3d83f5ba9066178f135fb/python/cudf/setup.py#L17-L27

It seems these are used in build time, though. Will try to dig the runtime ones.

1reaction

leofangcommented, Mar 24, 2020

For discussions on potential changes in cupy-feedstock, I’m redirecting to conda-forge/cupy-feedstock#46.

Top Results From Across the Web

how to set the cuda path in the conda environment? [closed]

You can set the CUDA_HOME environment variable with this command. ... this command to find the path of your conda-installed CUDA library.

Managing CUDA dependencies with Conda | by David R. Pugh

You can use the conda search command to see what versions of the NVIDIA CUDA Toolkit are available from the default channels.

Where is CUDAToolKit path when installed via conda?

I installed my PyTorch 1.0 using the command conda install pytorch torchvision cudatoolkit=9.0 -c pytorch while my system has an existing cudatoolkit ......

NVIDIA CUDA Installation Guide for Linux

The installation instructions for the CUDA Toolkit on Linux. ... If you do not see any settings, update the PCI hardware database that...

Install conda and set up a Pytorch 1.7, CUDA 11.1 ...

The first line creates our environment called “PyTorch” and you can select the python version (I choose version 3.7). · The second line...