Watson Studio にはデフォルトで Python 環境 Runtime 24.1 on Python 3.11 が組み込まれています。

一方、GPUを使用したい場合（GPUを使用したJupyter Notebookを使用したい場合など）に必要なサービス Runtime 24.1 on Python 3.11 for GPU はデフォルトではインストールされていません。使用したい場合は別途インストールする必要があります。Runtime 24.1 on Python 3.11 for GPU をインストールしたうえで、新しい GPU 環境テンプレートを作成すると、GPU環境を使用できるようになります。

デフォルトの状態

デフォルトの状態では、GPU環境のテンプレートは存在せず、新規テンプレート からも GPU構成は選べません。

Runtime 24.1 on Python 3.11 for GPU のインストール

手順はこちらですが、IBM Software Hub の他のサービスと同様のインストール手順でインストールできます。

＊インストールコマンドとしては、次のとおり components として、ws_runtimes を指定し、--extra-vars="cr_name=ibm-cpd-ws-runtime-241-pygpu" を指定しています。

cpd-cli manage apply-cr \
--components=ws_runtimes \
--release=${VERSION} \
--cpd_instance_ns=${PROJECT_CPD_INST_OPERANDS} \
--extra-vars="cr_name=ibm-cpd-ws-runtime-241-pygpu" \
--license_acceptance=true

この結果、cpd-cli manage get-cr-status で確認すると、デフォルトのRuntimeに加えて、GPU使用のRuntimeが追加されています（最終行）。

Component    CR Kind          CR Name                       Creation Timestamp    Namespace    Expected Version    Reconciled Version    Operator Information    Progress    Progress Message               Reconcile History    Status
-----------  ---------------  ----------------------------  --------------------  -----------  ------------------  --------------------  ----------------------  ----------  -----------------------------  -------------------  ---------
ws_runtimes  NotebookRuntime  ibm-cpd-ws-runtime-241-py     2025-02-17T01:44:59Z  wx           10.0.0              10.0.0                10.0.0034               100%        Last reconciliation succeeded  N/A                  Completed
ws_runtimes  NotebookRuntime  ibm-cpd-ws-runtime-241-pygpu  2025-02-27T00:48:38Z  wx           10.0.0              10.0.0                10.0.0034               100%        Last reconciliation succeeded  N/A                  Completed

GPU 環境テンプレートの作成

サービスのインストール後、新規テンプレート 作成画面では、Type で GPU を選択できるようになっています。GPUを選択すると、自動的にSoftware version に GPU Runtime 24.1 on Python 3.11 が選択されます。

テンプレート作成後、Notebook作成画面で、このランタイムを選択し使用できるようになります。

参考