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을 설치해주세요.
실시된 환경
- 먼저 도커 내부에 nvidia/cuda:10.2-cudnn7-devel-ubuntu18.04인 image를 pull 받습니다.
- 컨테이너를 열어준 뒤 하단의 방법으로 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로 대체
- 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 수정중, 추후 업데이트 예정)
나중에 돌릴 때 에러가 났을 때 자체적으로 참고하기 위한 용도로 작성된 포스팅이며 궁금한 부분이 있다면 댓글 남겨주시면 감사하겠습니다.
Author And Source
이 문제에 관하여(MeInGame env setting), 우리는 이곳에서 더 많은 자료를 발견하고 링크를 클릭하여 보았다 https://velog.io/@ghgh5317/MeInGame-env-setting저자 귀속: 원작자 정보가 원작자 URL에 포함되어 있으며 저작권은 원작자 소유입니다.
우수한 개발자 콘텐츠 발견에 전념 (Collection and Share based on the CC Protocol.)