报错信息:
Traceback (most recent call last): File “pruning.py”, line 5, in
import torch File
“/home/uriky/anaconda3/envs/compression/lib/python3.8/site-packages/torch/init.py”,
line 290, in from torch._C import * # noqa:
F403ImportError:
/home/uriky/anaconda3/envs/compression/lib/python3.8/site-packages/torch/lib/…/…/nvidia/cusparse/lib/libcusparse.so.12:
symbol __nvJitLinkAddData_12_1 version libnvJitLink.so.12 not defined
in file libnvJitLink.so.12 with link time reference
PyTorch 依赖的 NVIDIA cuSPARSE 库(libcusparse.so.12)在尝试访问 NVIDIA JIT Link 库(libnvJitLink.so.12)中的一个符号时失败了。这通常是因为 CUDA 或相关 NVIDIA 驱动版本与 PyTorch 版本不兼容所导致的。
如果你已经安装了最新版本的 CUDA 和驱动,但仍然遇到问题,尝试安装一个与你的 CUDA 版本兼容的 PyTorch 版本。使用 conda 或 pip 来安装特定版本的 PyTorch。使用 conda 安装 PyTorch 时,可以指定 CUDA 版本,如:
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
注意将 11.3 替换为你的 CUDA 版本。
cuda版本查询命令如下
nvcc -V