以前は、DETR 関連プログラムはサーバー上で実行されていました。サーバーは Linux を使用しているため、操作は比較的簡単ですが、簡単なデバッグが必要な場合は、サーバーを使用する必要はありません。今日は、Winodws で DETR プロジェクトをデバッグします。ここでは、Deformable-DETR と DAB-DETR を例として取り上げます。
最初は DAB-DETR の構成です。この部分は比較的単純です。DAB-DETR は CUDA オペレーターを構成する必要がないことに注意してください。DAB-DETR プログラムを実行するだけの場合は、インポート モジュールを削除できます。 DAB-Deformable-DETRの。
また、model/init の下にあるこのコードをコメントアウトします。
conda環境を作成する
conda create -n deformable_detr python=3.7
conda環境をアクティブ化する
activate deformable_detr
CUDA のバージョンを確認します。環境変数の CUDA の順序を切り替えて、対応する conda 環境を作成すると、この CUDA は conda 環境に従います。
nvcc --version
pytorch によると、CUDA バージョンに従って pytorch を選択すると、一致する cuda と pytorch は conda コマンドを使用して見つけることができないことがわかりました。そのため、pip を使用してメソッドに従うことができます。これは、pytorch と同時に小さな cuda を追跡できます。 。
conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.6 -c pytorch -c conda-forge
もちろん、以下も使用できます。
pip install torch==1.12.0+cu116 torchvision==0.13.0+cu116 torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu116
次に、プロジェクトのルート ディレクトリに切り替えて、requirement.txt 内の対応するパッケージをたどると、pycocotools のインストールが失敗し、Windows バージョンが直接インストールできることがわかります。
pip install pycocotools-windows -i https://pypi.tuna.tsinghua.edu.cn/simple
DAB-Deformable-DETR 追加構成
DAB-DETR の作者は Deformable-DETR をベースにして DAB-Deformable-DETR を開発しましたが、Deformable-DETR を実行するには追加のCUDAオペレーターが必要です。以下は構築手順ですが、先ほど作成した環境を基に以下の操作を行っていきます。
次に、model/ops フォルダーに切り替えて CUDA をコンパイルし、エラーを報告します。
「MultiScaleDeformableAttendant」拡張機能の構築エラー: Microsoft
Visual C++ 14.0 以降が必要です。「Microsoft C++
Build Tools」で入手します:
https://visualstudio.microsoft.com/visual-cpp-build-tools/
インターネット上では Visual C++ 14.0 をインストールする方法がありますが、このブロガーは時間と手間がかかりすぎると感じています。conda 環境に対応するコンパイル済みファイルを用意するだけでよく、ダウンロードしてインストールする必要はありません。巨大な Visual C++ 14.0。
conda install libpython m2w64-toolchain -c msys2
CUDA オペレーターを再度コンパイルすると、まだエラーが報告されていることがわかります。それを行う方法はありません。正直にインストールすることしかできません。ここでは、Visual Studio2017 バージョンをダウンロードすることをお勧めします。ブロガーは以前に 2022 バージョンをインストールしましたが、失敗しました。ダウンロードアドレスは以下の通りです。
https://visualstudio.microsoft.com/zh-hans/vs/older-downloads/
次に、インストール場所と必要なバージョンを選択します。
その後の CUDA コンパイル:
その後、プロジェクトの実行中にエラーが報告されます。
ファイル「D:\Anaconda\envs\deformable_detr\lib\shutil.py」、121 行目、fdst として open(dst, 'wb') を指定した
copyfile内: PermissionError: [Errno 13] 権限が拒否されました: 'C:\Users\ PENGXI~1\AppData\Local\Temp\tmp0srtpk4p\tmpqq71knx5.py'
このファイルを見つけてください
次のコード行を追加します。
それからまた走ります。OK
