항상 최신의 PC를 이용할 수는 없는 법. 오래된 PC에 PyTorch를 PIP를 이용해서 설치하고 나서, 실행을 하면 오류가 발생한다.
[W NNPACK.cpp:80] Could not initialize NNPACK! Reason: Unsupported
hardware.AVX2 명령어를 지원하지 않는 CPU에서 해당 오류가 발생한다고 한다. 구글 검색을 해보면 PyTorch를 소스로 설치하면서 USE_NNPACK=0으로 설치하면 된다고 한다.
USE_NNPACK=0 python3 setup.py install그런데 문제는 지금부터다. Tesla K40c 은 매우 오래된 아키텍쳐 기반이다. 그래서 추가 옵션이 필요하다. Tesla K40c는 3.5라고 한다.
TORCH_CUDA_ARCH_LIST="3.5" USE_NNPACK=0 python setup.py install그 다음에 딥러닝 학습에 필요한 여러 패키지를 설치한다. 대부분 torchvision을 이용할 텐데, torchvision도 소스로 설치해야 한다. 혹시나 싶어서 pytorch와 같은 방법으로 설치했다.
git clone --recursive https://github.com/pytorch/vision
cd vision
TORCH_CUDA_ARCH_LIST="3.5" USE_NNPACK=0 python setup.py install재현이 될지는 모르겠다.
NNPACK은 경고 메세지는 뜨지만, 일단 지나간다.
YOLOv5가 실행되기 시작했다.
