はじめに
DeepLabCut 2.1 をGPUを積んだ Windows 10 で使用する方法です。以下の公式ホームページと公式が作成したGitHubを参考にしたので、Ubuntu18.04を使う場合はこちらを参照してください。Python環境やドライバ等のインストールは済んでいれば必要ありません。
DeepLabCut 公式ホームページ
http://www.mousemotorlab.org/deeplabcut
GitHub
https://github.com/AlexEMG/DeepLabCut/blob/master/docs/installation.md
導入準備
- Anacondaを使ってPython 3.7をインストール https://www.anaconda.com/distribution/
- NVIDIAドライバをインストール
- CUDAインストールキットをインストール
- 以下のyamlファイルをダウンロード http://www.mousemotorlab.org/s/dlc-windowsGPU-aa86.yaml (Windows 10 GPU)
- Anaconda を管理者権限で実行 #環境構築
- 導入準備でダウンロードしたyamlファイルディレクトリを選択し仮想環境を作成
conda env create -f C:\yourenvironment\dlc-windowsGPU.yaml
- 仮想環境に入る
conda activate dlc-windowsGPU
- pipで以下3つをインストール
pip install numpy==1.17.2
pip install deeplabcut
pip install imgaug
- pythonを実行しdeeplabcutをインポートできれば成功!!
python
import deeplabcut