如果causal_conv1d_cuda是一个自定义的CUDA扩展或库,并且它是针对特定版本的CUDA编译的,那么你需要确保它与你的CUDA环境兼容。如果它不兼容,你可能需要重新编译它,或者找到一个与你的CUDA版本兼容的预编译版本。 重新编译通常涉及以下步骤: 确保你有正确的CUDA源代码。 设置正确的CUDA编译环境(包括环境变量和编译器路...
根据提供的信息,您已经成功安装了 causal-conv1d-cuda.cpython-310-x86-64-linux-gnu.so 文件。如果在安装过程中遇到 ImportError 错误,说明在您的系统中找不到这个文件的符号定义。为了解决这个问题,您可以按照以下步骤操作: 1. 确保您下载的安装包是完整的,没有缺失任何文件。 2. 检查您的系统路径是否正确,...
causal_conv1d_interface import causal_conv1d_fn, causal_conv1d_update 4 changes: 0 additions & 4 deletions 4 setup.py Original file line numberDiff line numberDiff line change @@ -125,10 +125,6 @@ def append_nvcc_threads(nvcc_extra_args): ext_modules = [] if not torch.cuda.is_...
Dao-AILab / causal-conv1d Public Notifications Fork 61 Star 335 Code Issues 16 Pull requests 7 Actions Projects Security Insights New issue TORCH_CUDA_ARCH_LIST support #39 Open bhack opened this issue Nov 18, 2024· 1 comment ...
cuda-version: '12.2.0' cuda-version: '12.2.2' - torch-version: '2.0.1' cuda-version: '12.2.0' cuda-version: '12.2.2' steps: - name: Checkout 2 changes: 1 addition & 1 deletion 2 causal_conv1d/__init__.py Original file line numberDiff line numberDiff line change @@ -1,3...