[初心者向け]深層学習(Python+GPU)の環境構築チュートリアル！！

はじめに

最近知り合いや研究室で深層学習の環境構築を教えてと言われることが多かったので，1つの記事にまとめようと思いました！！
windowsでpythonのインストールからGPUを使って学習できる環境を構築する他，Anacondaの仮想環境を用いたpython環境構築して物体検出のコードを動かすまでの一連の流れをまとめたいと思います！

これから研究などで深層学習を扱うけど何していいか分からない！ って人に読んでもらえると幸いです．
今回の記事ではpython+GPUの環境構築をざっくりとですが解説します！

その１ Python,Gitのインストール

こちらのサイトからAnacondaをダウンロードします

インストーラを起動するといろいろ出てきますが，基本的にはそのままNextでかまいません
ポチポチしているとインストールが完了します．

次にこちらからGitをインストールします

こちらもインストーラを起動するといろいろ出てきますが，基本的にはそのままNextでかまいません
ポチポチしているとインストールが完了します．

その２ CUDA,cuDNNのインストール

そもそもCUDA,cuDNNって何ぞやって人，ざっくりいうとこれはGPUを用いて機械学習を行う場合に入れなくてないけないツールです．(雑)

Visual Studio Communityのインストール

CUDAを動かすにはVisual Studioのコンパイラが必要なためインストールしていきます．
こちらのサイトからインストールしてください．
ついでに一番右のVisua Studio Codeはあとで絶対使うのでここでインストールしておいていいと思います

インストール時に必ずC++によるデスクトップ開発にチェックを入れて下さい．

CUDA Toolkitのインストール

こちらのサイトからCUDA Toolkitをダウンロードします．

今回はCUDA Toolkit 11.1.1をインストールします

プラットフォームを選択してインストーラをダウンロードします．
インストーラを起動するといろいろ出てきますが，基本的にはそのままNextでかまいません
ポチポチしているとインストールが完了します．

インストールが完了したら以下のコマンドでPATHが通ってるか確認してください．

nvcc -V

以下のような結果が出ると思います．

もしエラーなどが出る場合はPATHが通っていない可能性があります．
もしその場合はこちらの記事などを参考に環境変数のPATHを通してみて下さい．

cuDNNのインストール

こちらのサイトからcuDNNをダウンロードします
Download cuDNNを選択してください．
ダウンロードする際，アカウント登録をしないとダウンロードできない為，初めての方は登録をお願いします．(結構めんどいです)

Archived cuDNN Releasesをクリックします

今回はcuDNN8.1.1 for CUDA11.0 11.1 and 11.2をダウンロードします

ダウンロードし，解答したらcudaフォルダの中身を
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1にコピーします。

次にPATHを通します
まずシステム環境変数を開きます

環境変数のをクリックすると以下のようなウィンドウが出てきます
システム環境変数の新規をクリック

出てきたウィンドウに
変数名:CUDNN_PATH
変数値:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1
と入力します

以上でCUDA,cuDNNのインストールは完了です．

GPUが使えるか確認

ちゃんと使えるようになっているか確認しましょう
windowsのスタートメニューからanacondaプロンプトを開きます

次にanacondaプロンプト上で以下のコマンドを打って深層学習ライブラリであるpytorchをインストールしてみましょう

pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

インストールが完了したら以下のpythonファイルを作成します

test.py

import torch
print(torch.cuda.is_available())

このpythonファイルをつくったディレクトリまで移動し，以下のように実行してTrueが帰ってくれば無事にGPUでの開発環境が完成です！

終わりに

いかがだったでしょうか？
僕が機械学習を始めた時はこの環境構築に詰まってかなりの時間を費やした記憶があります．
この記事で環境構築が楽にできるようになってくれると幸いです．

次はいよいよ物体検出のコードを動かします！
是非ご覧ください！

参考