MeInGame env setting

2325 ワード

https://github.com/FuxiCV/MeInGameコードを迂回するために,環境設定を行い,大量のエラーに対して成功した方法をまとめることを試みた.

インストールするライブラリ


pytorch、pytorch 3 d、tentorflow-gpuをインストールする必要があります.
pytorch 3 dをインストールする環境については、pytorch3dのリンクを参照してください.
多分持ってきたら、

pythonバージョンとpytorchバージョンを組み合わせたいのですがtentorflow-gpu=1です.Xをインストールするには、pythonが3.7 versionに達する必要があります.この点に注意し、Pythonをインストールしてください.

実装済み環境

  • まず、nvidia/cuda:10.2-cudn 7-devel-ubuntu 18をドッキングステーション内部に追加します.04の画像を取得します.
  • コンテナを開き、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でこれらのライブラリをインストールできます.
  • 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
    すべてのインストールが完了し、コードを実行するときにテンソルストリームのgpu使用量がfalseになる可能性があります.
    **テンソルストリームGPU使用確認コード**
    import tensorflow as tf
    tf.test.is_gpu_available()
    **pytorch gpu使用確認コード**
    import torch
    torch.cuda.is_available()
    私が確認したとき、テンソルストリームにfalseが現れ、ブログを参照して、gpuがテンソルストリームから戻ることができるようにしました.
    では、インストール後にコードを回して戻ります!(現在cuda out memory errorを変更しており、後で更新します)
    後で回った時に間違ったことがあった時に自分の参考にするために書いたヒント、気になることがあったらメッセージありがとうございます.