在尝试使用SAM2时,根据CUDA 12.1版本安装了pytorch。
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
版本为torch 2.4.0+cu121 torchaudio 2.4.0+cu121 torchvision 0.19.0
尝试import torch
时,报错如下:
OSError: [WinError 126] 找不到指定的模块。 Error loading "D:\anaconda\envs\py311\Lib\site-packages\torch\lib\fbgemm.dll" or one of its dependencies.
原因及解决办法:
\torch\lib\fbgemm.dll 目录下缺少 libomp140.x86_64.dll 文件。
下载链接:https://www.dllme.com/dll/files/libomp140_x86_64/00637fe34a6043031c9ae4c6cf0a891d/download
解压后,把 libomp140.x86_64.dll 复制到 fbgemm.dll 同目录下就行了。