OpenBMB / VisCPM

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

报错 AssertionError: Torch not compiled with CUDA enabled #36

Closed sudnmf closed 4 months ago

sudnmf commented 9 months ago

D:\Anaconda\envs\viscpm\lib\site-packages\torchvision\models\detection\anchor_utils.py:63: UserWarning: Failed to initialize NumPy: module compiled against API version 0x10 but this version of numpy is 0xf (Triggered internally at ..\torch\csrc\utils\tensor_numpy.cpp:77.) device: torch.device = torch.device("cpu"), Traceback (most recent call last): File "D:\ViscpmProj\VisCPM\hello.py", line 5, in viscpm_chat = VisCPMChat(model_path, image_safety_checker=True) File "D:\ViscpmProj\VisCPM\VisCPM\viscpm_chat.py", line 40, in init self.cpm_model = CPMBeeTorch(self.config) File "D:\ViscpmProj\VisCPM\VisCPM\models\cpmbee.py", line 74, in init self.input_embedding = EmbeddingExt( File "D:\ViscpmProj\VisCPM\VisCPM\models\modules\embedding.py", line 77, in init self.rotary_emb = RotaryEmbedding( File "D:\ViscpmProj\VisCPM\VisCPM\models\modules\position_embedding.py", line 222, in init base ** (torch.arange(0, dim, 2, device="cuda", dtype=torch.float32) / dim) File "D:\Anaconda\envs\viscpm\lib\site-packages\torch\cuda__init__.py", line 221, in _lazy_init raise AssertionError("Torch not compiled with CUDA enabled") AssertionError: Torch not compiled with CUDA enabled

JamesHujy commented 7 months ago

能提供一下具体的环境配置吗?