[ Python ] Alphapose 환경세팅 ( Window OS )

Desktop spec :

OS : Windows 11 Pro 64x
CPU : Ryzen9 5900x
Ram : 32GB
GPU : RTX3070 Ti

Cuda Version : 11.8

Pytorch 및 기타 모듈:
torch 2.1.2+cu118
torchaudio 2.1.2+cu118
torchvision 0.16.2+cu118

opencv-contrib-python 4.8.0.74
opencv-python 4.9.0.80 onnx-simplifier 0.4.10
onnxruntime 1.15.1
PyYAML 6.0.1

Cuda Toolkit : 11.8v
Cudnn :
CUDA Toolkit에 있는 폴더 내의 bin, include, lib 폴더에 cuDNN에서 다운로드한 bin, include, lib 폴더 내의 파일들을 추가

cuDNN 버전 8 이전
C: > Program Files >NVIDIA GPU Computing Toolkit > CUDA > v11.8 (설치되어있는 CUDA 버전) > include > cudnn.h
cuDNN 버전 8 이후
C: > Program Files > NVIDIA GPU Computing Toolkit > CUDA > v11.8 (설치되어있는 CUDA 버전) > include > cudnn_version.h

위부분이 설치를 끝냈다면, 아래 명령어로 실행을 한다

> python

> import torch
> torch.cuda.is_available()

True

x = torch.rand(3,5)
print(x)

tensor([[0.8249, 0.2061, 0.7752, 0.2189, 0.6357],
[0.4776, 0.0924, 0.1574, 0.0692, 0.9763],
[0.9525, 0.3279, 0.7151, 0.1584, 0.5715]])

정상작동하는 것을 확인 했다.

================== Alphapose ======================
1.Alphapose Git의 오픈소스를 사용 하였고, Pose Estimate 중에서 퍼포먼스가 월등하여 사용

AlphaPose/docs/INSTALL.md at master · MVIG-SJTU/AlphaPose

Real-Time and Accurate Full-Body Multi-Person Pose Estimation&Tracking System - MVIG-SJTU/AlphaPose

github.com

Real-Time and Accurate Full-Body Multi-Person Pose Estimation&Tracking System - MVIG-SJTU/AlphaPose

AlphaPose/docs/INSTALL.md at master · MVIG-SJTU/AlphaPose

Real-Time and Accurate Full-Body Multi-Person Pose Estimation&Tracking System - MVIG-SJTU/AlphaPose

github.com

2.CUDA , CUDA_HOME을 각각 환경변수를 지정해줘야하고, Windows의 Visual studio 를 사용하기 때문에 이에 대한 CL 을 환경변수를 지정해줘야한다

CUDA_HOME
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8

CUDA_PATH
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin

시스템 환경변수부분도 편집

Visual Studio 2022 에대한 CL 및 라이브러리 환경변수
...\Microsoft Visual Studio\2022\Community\VC\Tools\MSVC\<version>\bin
...\Microsoft Visual Studio\2022\Community\VC\Tools\MSVC\<version>\lib
...\Microsoft Visual Studio\2022\Community\VC\Auxiliary\VS\include
...\Microsoft Visual Studio\2022\Community\VC\Redist\MSVC\<version>

Cuda에 관련한 라이브러리 환경변수 
...\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin
...\NVIDIA GPU Computing Toolkit\CUDA\v11.8\lib
...\NVIDIA GPU Computing Toolkit\CUDA\v11.8\include
...\NVIDIA GPU Computing Toolkit\CUDA\v11.8\include\CL
...\NVIDIA GPU Computing Toolkit\CUDA\v11.8\libnvvp

3. setup.py 부분에서

force_compile 부분이 False가되어있는데 윈도우 사용자는 True로 변경