6
16

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?

More than 1 year has passed since last update.

Jupyter NotebookでローカルのGPUを使う方法

Posted at

はじめに

Jupyter NotebookでローカルPCのGPUを使ってみたいなぁ・・・せっかくワークステーション持ってるんだし。
というのが今回のモチベーション。

やったこと

とにかくやったことを順番に解説していきます。

①Anacondaで仮想環境を構築

スタートメニューからAnaconda Navigatorを起動。
image.png

Environments→Createをクリック。
image.png

任意の名前を入力(今回は、"local_GPU"とした。)して、Createボタンをクリック。
image.png

仮想環境(local_GPU)が作成されました!
image.png

②CUDA toolkitをインストール

下記のサイトから最新のインストーラをダウンロードします。
https://developer.nvidia.com/cuda-toolkit-archive
image.png

PCの環境に合わせて選択し、Downloadをクリック。
image.png

ダウンロードした実行ファイルを実行してインストーラを起動。

そのままOK
image.png

同意して続行するをクリック。
image.png

高速(推奨)を選択して次へ
image.png

インストールが始まりますので、しばらく待ちます。
image.png

次へ
image.png

閉じる
image.png

③cuDNNをインストール

下記のサイトから最新のものをウンロードします。
https://developer.nvidia.com/rdp/cudnn-archive

右上のJoinをクリック。
image.png

メールアドレスを入力してNEXT
image.png

・アカウント登録していない場合
パスワードを入力してアカウントを作成する画面が表示されますので、画面に従ってアカウント登録をします。
image.png

・既にアカウント登録している場合
今はしないをクリック
image.png

認証後、インストーラを選択してダウンロードします。
今回は、"Download cuDNN v8.6.0 (October 3rd, 2022), for CUDA 11.x"を選択しました。
image.png

ZIPファイルでダウンロードされるので、解凍します。
(ちなみに解凍した中身はこんな感じです。)
image.png

次に、CUDA toolkitをインストールしたディレクトリ(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.0)に解凍したファイルをすべてコピーします。

・bin
image.png

・include
image.png

・lib
image.png

④zlibをインストール

下記のサイトを開き、ZLIB.DLLを右クリック→名前を付けてリンク先を保存をクリック。
https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html#install-zlib-windows
image.png

ZIPファイルでダウンロードされるので、解凍します。
(ちなみに解凍した中身はこんな感じです。)
image.png

zlib123dllx64\dll_x64のzlibwapi.dllをCUDA toolkitをインストールしたディレクトリ(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.0\bin)にコピーします。
image.png

⑤Tensorflow-gpuをインストール

Anaconda Navigatorより、先ほど作成した仮想環境(local_GPU)からTerminalを開きます。

image.png

下記のコマンドを実行します。

pip install tensorflow-gpu==2.6.0

インストールが完了したら、下記のコマンドを実行します。

>python
>>> from tensorflow.python.client import device_lib
>>> device_lib.list_local_devices()

実行結果がこんな感じになっていればOKです。
image.png
↑↑device_type:GPUってのが表示されればOKですね。

⑥Jupyter Notebookに仮想環境を追記

仮想環境(local_GPU)のTerminal上で下記のコマンドを実行します。

pip install ipykernel

インストールが完了したら、さらに下記のコマンドを実行します。
xxxxxの部分は仮想環境名(今回の場合は"local_GPU")に置き換えてください。

ipython kernel install --user --name=xxxxx

Kernelに仮想環境が追加されているのが確認できます。
image.png

以上!

おわりに

上から順番にやっていけば環境構築できるようにまとめたつもりです。
私が実践したときはたくさんのエラーと出会い、いろんなサイトを調べながら行ったり来たりで大変でした・・・。

みなさんの一助となれば幸いです!

6
16
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
6
16

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?