Dao-AILab/causal-conv1d

installation for nvcr images

ftgreat opened this issue · 2 comments

ftgreat commented

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

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.

  • ❤️1
ftgreat commented

thanks.
installed succefully from source build with force build.