OpenMythosのサンプルプログラムを動かしました

1.概要 前回、OpenMythosの環境構築まで実現したので、GitHubにあるサンプルプログラム(Usage)を動かしてみました。一部エラーが発生しましたが、結果出力部分なので修正をしました。その内容を記述します。 2.詳細 Usageに記載されているPythonプログラムをopenmythos_usage.pyとして作成します。 $ source ~/mypy/bin/activate (mypy) $ pythoh3 openmythos_usage.py 下記エラーとなります [MLA] Parameters: 1,538,626 [MLA] Logits shape: torch.Size([2, 16, 1000]) [MLA] Generated shape: torch.Size([2, 24]) Traceback (most recent call last):   File "/home/nakasima/openmythos/openmythos_usage.py", line 48, in <module>     rho = torch.linalg.eigvals(A).abs().max().item()           ^^^^^^^^^^^^^^^^^^^^^^^ RuntimeError: linalg.eig: The input tensor A must have at least 2 dimensions. コードの下記部分を修正します #rho = torch.linalg.eigvals(A).abs().max().item() rho = A.abs().max().item() $ source ~/mypy/bin/activate (mypy) $ pythoh3 openmythos_usage.py [MLA] Parameters: 1,538,626 [MLA] Logits shape: torch.Size([2, 16, 1000]) [MLA] Generated shape: torch.Size([2, 24]) [MLA] Spectral radius ρ(A) = 0.3679 (mus...

Code Llamaはpytorch環境でGPU必須でした

1.概要

知人からCode LlamaはGPUがない環境で動作するのですか? と問合せを受けました。そこで、GPUを利用しないpytorch環境を構築して試験しました。結果は、CUDAがないと動作しないので、nvidaのGPUは必須のようです。この内容を記述します。

2.詳細

(a) 実行環境
(b) docker + pytorchの環境構築
(c) Google Colabで実行された内容の再現

(a) 再現環境
利用したPCは下記スペックです
CPU: i3-7100
GPU: nvidia 1050Ti
Memory: 16GB
Disk: 500GB(Hard disk)
OS: Ubuntu-20.04.6(desktop)

(b) docker + pytorchの環境構築
ubuntu-20.04.6環境をBackupからリストアしてクリーン環境にして実行。
nvidiaのドライバーは、nouveauです。
docker hubのcontinuumio/miniconda3:latestを利用
pytorchは、Stable(2.0.1)利用

(c) Google Colabで実行された内容の再現
実行はjupyter-notebookで実行しました。
パッケージのインストールは、実行できました。
モデルとトークナイザーの準備も実行できました。
モデルとトークナイザーの準備の下記部分でエラーとなりました。
inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
エラー内容は下記メッセージです。
AssertionError: Torch not compiled with CUDA enabled

3.所見
上記エラー部分の .to("cuda")を削除して実行してみましたが、無駄な抵抗でした。

参考
[外部サイト参照]
Google Colab で Code Llama を試す
Llama 2 learns to code

コメント

このブログの人気の投稿

miniPCのBMAXでWindows11のsecure boot設定漏れでトラブル発生

LinuxMint-22.3にWine 11.6をサクッと入れてみました

LinuxMint 21.3にWinBoat導入を試してみました