mmsegmentation

trouble shooting

OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.

mmsegmentation을 사용하기 위해 mmcv와 mmcv-full을 pip install 할 때 mmcv는 정상적으로 설치되었으나 mmcv-full를 설치할 때 다음과 같은 에러가 발생했습니다. mmcv github에서 readme를 확인해보니 -> 링크

pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.9.0/index.html

위의 코드를 사용하라고 되어있어 그대로 입력해보니 정상적으로 설치되었습니다.

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

pip uninstall mmcv-full
pip uninstall mmsegmentation

pip install openmim
mim install mmsegmentation

mim으로 설치해야 에러가 안 납니다.
공식 github의 get_started.md를 읽고 설치했는데 왜 에러가..?
openmmlab에서 제공하는 패키지들은 mim으로 설치해야 하는걸까요..

CUDNN_STATUS_NOT_INITIALIZED

model norm_cfg의 type을 SyncBN -> BN으로 수정해주면 됩니다.