[YOLO] ModuleNotFoundError: No module named 'torch'

Jupyter Notebook으로 YOLO 모델을 돌리려는데 다음과 같은 오류가 떴다.
이때는 일단, torch가 다운되어있지 않다면 torch 먼저 다운로드 해준다.

나의 경우에는 기존에 깔린 torch를 지우고 다시 깔아야 했기에 삭제 후 강제 설치했다.

pip uninstall torch
pip uninstall -y -q torch
pip install torch --force-reinstall

이 코드는 아나콘다 터미널에서 사용했다.

그 후, Jupyter에서 다음과 같은 코드로 확인해준다.

import torch
print("PyTorch 버전:", torch.__version__)
print("GPU 사용 가능:", torch.cuda.is_available())

이 때 제대로 설치되었다면

이런 식으로 결과가 나온다.

그리고 발생했던 오류를 잡기 위해 Python 경로도 확인해준다.

import sys
print(sys.executable)

결과
!<여기에 나온 경로> train.py --data ...

이런 식으로 Python의 경로가 나오는데 여기에 나온 경로로 YOLO 모델을 돌려주면 된다.

예시 코드는 다음과 같다.

!<경로>/python -u train.py \
  --img 640 \
  --batch 16 \
  --epochs 20 \
  --data <경로>/data.yaml \
  --weights yolov5s.pt \
  --name yolo_try1