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
コメント
コメントを投稿