MeInGame env setting

https://github.com/FuxiCV/MeInGame 해당 코드를 돌리기 위해 환경세팅을 실시하였고, 수 많은 에러들을 마주해서 성공한 방법을 정리해보려고 합니다.

필수로 설치되어야 하는 라이브러리

pytorch, pytorch3d, tensorflow-gpu가 설치되어야합니다.
pytorch3d를 설치하기 위한 환경은 pytorch3d 이 링크를 참고해주세요.
대략적으로 가지고 오자면,

python version 과 pytorch version을 맞춰주려고 하는데 tensorflow-gpu=1.X를 install하려면 python이 최대 3.7version까지여야 합니다. 이 점을 유의하시어 python을 설치해주세요.

실시된 환경

  1. 먼저 도커 내부에 nvidia/cuda:10.2-cudnn7-devel-ubuntu18.04인 image를 pull 받습니다.
  2. 컨테이너를 열어준 뒤 하단의 방법으로 python과 pip를 설치해줍니다.
apt update
apt install python3 python3-distutils -y
일단 저 2개의 패키지를 설치해준다.

python3 --version으로 버전을 확인해보면 3.6.9인데,
저는 python 3.7을 설치하려고 해 하단의 방법을 사용했습니다..

curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py

--> curl이 없을 때 apt install curl
로 get-pip.py 를 다운 받고

python3 get-pip.py
-->3.6버전이 따로 있으니 3.6버전으로 맞추던, 아님 python upgrade하면 된다
	apt install python3.7
	python get-pip.py로 대체
  1. pip가 설치가 완료됐으니 이제 pip install을 통해 해당 라이브러리들을 install하면 됩니다.
pip install torch==1.10.1+cu102 torchvision==0.11.2+cu102 torchaudio==0.10.1 -f https://download.pytorch.org/whl/torch_stable.html
pip install opencv-python fvcore h5py scipy scikit-image dlib scikit-learn tensorflow-gpu==1.14.0 gast==0.2.2
pip install face-alignment==1.1.1
pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py37_cu102_pyt1100/download.html

모두 설치가 완료됐고, 아마 해당 코드를 실행하면 tensorflow에서 gpu 사용량이 false가 나오는 경우도 있습니다.

** tensorflow gpu 사용 확인코드 **
import tensorflow as tf
tf.test.is_gpu_available()
** pytorch gpu 사용 확인코드 **
import torch
torch.cuda.is_available()

저는 확인을 해봤을 때 tensorflow에서 false가 나와서, 해당 블로그 를 참조하여 tensorflow에서 gpu가 돌아갈 수 있도록 했습니다.

그럼 설치한 후 코드를 돌리면 돌아갑니다! (현재 cuda out of memory error 수정중, 추후 업데이트 예정)

나중에 돌릴 때 에러가 났을 때 자체적으로 참고하기 위한 용도로 작성된 포스팅이며 궁금한 부분이 있다면 댓글 남겨주시면 감사하겠습니다.

좋은 웹페이지 즐겨찾기