インストールの検証は、3つの重要なステップに分かれている:
- 基本的な環境検査::
- Python ≥ 3.10 と git がインストールされていることを確認してください。
- うごきだす
import torch; print(torch.cuda.is_available())
CUDAサポートの確認
- プロジェクト展開::
- とおす
git clone https://github.com/jingyaogong/minimind.git
ソースコードを入手する - Tsinghua Sourceを使用して、依存関係のインストールを高速化します:
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
- とおす
- 機能検証::
- 1.6GBをダウンロード
pretrain_hq.jsonl
データセット ./データセット - うごきだす
python eval_model.py --load 1
基本的な推理力のテスト
- 1.6GBをダウンロード
出遭うflash_attn
インストールに失敗した場合は、関連するコードを一時的にコメントアウトしてテスト実行することができます。
この答えは記事から得たものである。MiniMind:ゼロから2時間のトレーニング 26MパラメータGPTオープンソースツールについて