Windows에서 Pytorch3d를 설치하는 방법
내 노트북에 Windows GPU가 있는데 Pytorch3d를 설치하지 못해서 설치하는데 많은 노력이 필요했는데, 끈기를 기념하기 위해 모두와 공유합니다.
WIN에서 Pytorch3d 설치
1. 질문:
최근에 3D 재구성을 시작했을 때 많은 프로젝트 프로그램에 Pytorch3d 라이브러리가 필요하다는 사실을 발견했습니다. 공식 웹사이트에서는 이를 다음과 같이 정의합니다.
PyTorch3D는 PyTorch를 통해 3D 컴퓨터 비전 연구를 위한 효율적이고 재사용 가능한 구성 요소를 제공합니다.
Pytorch3d 라이브러리는 3D 시각 처리를 위한 다양한 구성 요소를 제공할 수 있음을 알 수 있는데, Ubuntu에서의 설치는 비교적 간단하지만 Win에서는 다소 번거롭습니다. 이번 포스팅은 개인적인 설치 방법에 대한 기록입니다.
추천 콘텐츠
2. 설치방법
설치 지점: cuda, pytorch 및 NVIDIA CUB 버전 간의 엄격한 대응
설치에 필요한 링크:
- Pytorch3d 버전: Pytorch3d 릴리스 다운로드
- NVIDIA CUB 다운로드: NVIDIA/cub 다운로드
- 위에서 언급한 PyTorch 공식 웹사이트 내역 버전 링크( Previous PyTorch Versions | PyTorch )에서 v1.11.0의 설치 명령을 찾으며, 여기에서 cuda11.3에 해당하는 PyTorch 설치 명령이 선택됩니다.
설치하기 전에 먼저 버전 대응을 확인해야 합니다. 제가 설치한 버전은 다음과 같습니다.
CUDA 툴킷 11.3 + pytorch 1.11.0 + Visual Studio 2019 + Pytorch3d 0.6.2 + Python 3.10
1. NVIDIA CUB의 경우 CUDA Toolkit과의 버전 관계는 다음과 같습니다.
제가 설치한 cuda 버전은 11.3이므로 CUB는 해당 버전인 1.11.0을 선택하였으며, 다운로드 후 압축을 푼 후 환경변수에 해당 경로를 추가해 주세요.
2. 종속성을 설치합니다.
conda install -c fvcore -c iopath -c conda-forge fvcore iopath
3. Pytorch3d 버전을 선택하십시오.Pytorch3d의 각 버전에는 아래와 같이 해당하는 pytorch 버전이 표시되어 있습니다.
위 그림을 보면 pytorch3d 0.6.2의 적용 가능한 Pytorch 버전은 1.10.1, 1.10.2, 1.11.0이고, 적용 가능한 Python은 3.10이나 Pytorch1.6.0과 Python3.6은 더 이상 지원하지 않는 것을 알 수 있다. 토치는 여전히 이것을 따릅니다. 해당하는 pytorch3d 버전을 다운로드하세요.
4. VS 2019를 설치한 후 아래와 같이 "x64 Native Tools Command Prompt for VS 2019" 터미널을 연 다음 pytorch3d가 압축 해제된 디렉터리 경로로 cd합니다.
순서대로 입력하세요:
Windows의 명령 프롬프트에서 환경 변수를 설정합니다.
DISTUTILS_USE_SDK=1
설정 PYTORCH3D_NO_NINJA=1 설정
Windows의 PowerShell에서 환경 변수를 설정합니다.
$env:DISTUTILS_USE_SDK = 1Linux 및 macOS에서 환경 변수를 설정합니다.
export DISTUTILS_USE_SDK=1
그런 다음 pytorch3d에서 setup.py 파일의 소스 코드를 수정합니다. extra_compile_args = {“cxx”: [“-std=c++14”]}를 다음으로 변경합니다. extra_compile_args = {“cxx”: []};
5. 마지막으로 "VS 2019용 x64 기본 도구 명령 프롬프트" 터미널에 다음을 입력합니다.
python setup.py 설치 --사용자
내 컴퓨터에서 설치 과정은 약 5분 정도 걸렸고 페이지가 계속해서 뛰어다니다가 마침내 성공을 거두었습니다.
그래도 설치에 실패하면 저에게 연락해주세요.