CUDA 11.6 Pytorch3d Install 설치 방법

 

https://github.com/facebookresearch/pytorch3d/blob/main/INSTALL.md

5시간을 왜 pytorch3d가 install이 안돼서 끙끙됐는데 알고보니 굉장히 간단한걸 깨달았다.

분명 위의 사이트를 그대로 따라하면 된다는 건 다들 알고있겠지. 하지만, 그대로 따라할 때 내가 놓친 부분이 있다. 아래에 설명하였다. pytorch3d를 깔려면 cuda10.1이 필요하다는 오해가 있을 수 있는데 사실 상관 ㄴㄴ함. CUDA11.6도 가능

 

 

<현재 나의 setting>

1. CUDA 11.6

2. Python 3.9

3. Pytorch 1.13.0

 

# Step1. 

위 사이트 그대로 다운받는다. conda create해서 도화지에서 시작하기

conda create -n pytorch3d python=3.9
conda activate pytorch3d
conda install pytorch=1.13.0 torchvision pytorch-cuda=11.6 -c pytorch -c nvidia
conda install -c fvcore -c iopath -c conda-forge fvcore iopath

# Step2

이 때 중요한건 경로 지정 cd ~ 새로운 폴더를 지정해서 아래의 코드를 실행한 후 그 폴더에 잘 다운받아졌는지 확인한다.

curl -LO https://github.com/NVIDIA/cub/archive/1.10.0.tar.gz
tar xzf 1.10.0.tar.gz
export CUB_HOME=$PWD/cub-1.10.0

 

# Step3

이 명령어만 치면 성공적으로 깔림. 

pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py38_cu113_pyt1110/download.html

이 때 중요한건 위에서 언급한  setting 3가지의 버전대로 아래 형광펜 친 부분을 수정해주는 것임.

<나의 setting>

CUDA 11.6 / Python 3.9 / Pytorch 1.13.0

 

끝.

이제 import pytorch3d하면 error 없이 잘 뜨는 걸 확인할 수 있다.

참고블로그

https://velog.io/@ghgh5317/pytorch3d-%ED%99%98%EA%B2%BD%EC%84%A4%EC%A0%95%ED%95%98%EA%B8%B0

pytorch3d env setting

---

문제 발생!

삽질 해결한듯 싶었으나 또 다시 삽질 시작,,,..절망스럽다.  python 3.9로 했더니 

/envs/pytorch3d/lib/python3.9/multiprocessing/resource_tracker.py:216: UserWarning: resource_tracker: There appear to be 11 leaked semaphore objects to clean up at shutdown warnings.warn('resource_tracker: There appear to be %d

문제 발생함.

구글링해보니 이게 파이썬 버전을 3.7로 다운그레이드 시키면 해결될거라 함.

그러면 cuda도 10.1 로 다운 그레이드해서 설치해야함.

다시 삽질 시작...,,,,,,빡침.