search

Dao-AILab / causal-conv1d

Causal depthwise conv1d in CUDA, with a PyTorch interface
BSD 3-Clause "New" or "Revised" License
335 stars 61 forks source link

installation for nvcr images #10

Closed ftgreat closed 11 months ago

ftgreat commented 11 months ago

I use the nvcr base image: nvcr.io/nvidia/pytorch:23.06-py3.

Pip install directly failed as follows. image

Should I install from sources and how to set these variables below in setup.py for nvcr base image?

image

Thanks for your advice.

tridao commented 11 months ago

nvcr uses pytorch nightly versions so it's a bit hard to make sure the compiled CUDA wheels are compatible. causal-conv v1.1.1 should work with nvcr 23.12.

ftgreat commented 11 months ago

thanks. installed succefully from source build with force build.