search

OpenBMB / VisCPM

[ICLR'24 spotlight] Chinese and English Multimodal Large Model Series (Chat and Paint) | 基于CPM基础模型的中英双语多模态大模型系列
1.06k stars 93 forks source link

CUDAContext.h:6:10: fatal error: cusparse.h: No such file or directory #5

Closed funykatebird closed 1 year ago

funykatebird commented 1 year ago

cuda11.6 python 3.10 In file included from /tmp/pip-install-docs1q76/bmtrain_cd370155ab894fa5969602d5d8168e37/csrc/nccl.cpp:4:0: /opt/conda/envs/viscpm/lib/python3.10/site-packages/torch/include/ATen/cuda/CUDAContext.h:6:10: fatal error: cusparse.h: No such file or directory

Cppowboy commented 1 year ago

您的cuda安装可能有问题，或者是未指定CUDA_HOME。建议您重新安装cuda，并指定CUDA_HOME环境变量后重新测试。

可以到https://developer.download.nvidia.cn/compute/cuda/repos/ubuntu2204/x86_64/ 下载对应版本的依赖安装。