1.问题描述

按照mmsegmentation官方安装脚本安装：

conda create -n open-mmlab python=3.8 -y
conda activate open-mmlab

conda install pytorch=1.6.0 torchvision cudatoolkit=10.1 -c pytorch
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.6.0/index.html
git clone https://github.com/open-mmlab/mmsegmentation.git
cd mmsegmentation
pip install -e .  # or "python setup.py develop"

运行时报错：

ImportError: libtorch_cuda_cu.so: cannot open shared object file: No such file or or directory

2.原因和解决方案

（1）先检查环境中是否存在libtorch_cuda_cu.so文件：

文件应该存在的路径：
anaconda3/envs/环境名称/lib/python3.8/site-packages/torch/lib

（2）若存在，证明是查找文件路径出现问题：

事实上，是mmsegmentation官方文档建议的环境名称open-mmlab包含短连接号“-”导致程序无法找到libtorch_cuda_cu.so文件

（3）建议采用open_mmlab等其他环境名称，避免使用“-”号，即可解决