<aside> 📌

절대로 Conda 환경에 OpenCV, CUDA, Pytorch를 다시 설치하지 마세요!! conda create 도 하지마세요!! 초보자에겐 safe_set_pytorch_conda.sh 로만 설치하는게 안전한 방법입니다!!!!

</aside>

😍 안전하고 빠르게 Conda 가상환경에 OpenCV+CUDA+Pytorch 설치하기!

Jetson의 환경은 일반적인 환경과 다른 아키텍처(aarch64)를 가지고 있어서 조심하지 않으면 다음과 같은 문제가 발생합니다.

• pip install 이나 conda install을 통해 패키지를 설치하면 설치가 되지 않음 (aarch64를 지원안함)
• GPU를 사용할 수 없는 버전으로 설치됨 (torch를 설치했는데 cpu버전 설치됨).

이러한 문제는 우리의 Jetson이 고정된 Python 버전, 고정된 CUDA 버전 그리고 고정된 Pytorch 버전을 사용하기 때문에 발생합니다. 구체적으로, Jetson에서는 다음 라이브러리들만 사용할 수 있습니다.

• Hardware: Jetson Xavier NX
• OS: Ubuntu 20.04 (L4T r35.6.x) / JetPack 5
• Python: 3.8
• Key libs:
  • CUDA 11.4 (JetPack)
  • cuDNN 8.x (JetPack)
  • PyTorch 1.11.0~2.1.0 (Jetson wheel)
  • Torchvision 0.11.1~0.16.1 (from source)

여기서, Pytorch와 Torchvision은 서로 호환되는 버전 페어가 있으므로 버전을 꼭 맞춰서 설치하여야 합니다. (버전 호환 Matrix)

거기다 우리는 앞에서 CUDA가 사용가능한 OpenCV를 빌드했으므로, 이를 가져와서 conda 환경에서도 사용가능하게 만들어야 합니다. 즉, conda 환경에서 opencv-python 을 다시 설치하는게 아니라, 원래 가지고 있는 cuda 지원 가능 opencv를 써야합니다.

이런 규약들을 전부 지켜가면서 하려면.. 너무 힘들기 때문에.. 안전하고 빠르게 Conda 가상환경을 만드는 쉘스크립트를 추가했습니다 😍!!

만약 conda 환경을 만들고 싶다면, 바로 아래 쉘 스크립트를 쓰면 됩니다! 아직 repo를 받지 않았다면 다음을 타이핑해서 repo를 내려받아 사용하세요.