git clone을 이용한 pytorch3d 설치

CV 관련 작업을 하다 보면 pytorch3d를 자주 접하는데, CUDA 툴킷 버전이나 deprecate된 함수 또는 최신 함수를 사용하려면 토치 버전이랑 다른 라이브러리의 requirements가 충돌하는 경우가 꽤 자주 있다.

공식 설치 가이드에 PyTorch 2.1.0~2, 2.2.0~2, 2.3.0~1, 2.4.0~1 이렇게 지원 버전이 적혀있지만 매번 버전 맞추는 것도 까다롭고 번거롭다.

예를 들어서 scaled_dot_product_attention 같은 함수는 2.0 이상의 토치를 요구하는데 도커 내 CUDA 툴킷이 11.7이고 토치 버전이 1.x인데다가 거기 맞춰서 환경이 구성돼 있으면 골치아프다.

지원 버전만 확인하면 git clone으로 해결할 수 있다.

git clone https://github.com/facebookresearch/pytorch3d.git
cd pytorch3d && pip install -e .

Ref.
1. CUDA 12 이상에서 pytorch3d 설치하기