A verificação da instalação é dividida em três etapas principais:
- Inspeção ambiental básica::
- Certifique-se de que o Python ≥ 3.10 e o git estejam instalados.
- estar em movimento
import torch; print(torch.cuda.is_available())Verificar o suporte a CUDA
- Implementação do projeto::
- aprovar (um projeto de lei ou inspeção etc.)
git clone https://github.com/jingyaogong/minimind.gitObter código-fonte - Use o Tsinghua Source para acelerar a instalação de dependências:
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
- aprovar (um projeto de lei ou inspeção etc.)
- Verificação funcional::
- Faça o download do arquivo de 1,6 GB
pretrain_hq.jsonlConjunto de dados para . /dataset - estar em movimento
python eval_model.py --load 1Teste de habilidades básicas de raciocínio
- Faça o download do arquivo de 1,6 GB
encontrarflash_attnSe a instalação falhar, você poderá comentar temporariamente o código relevante para uma execução de teste.
Essa resposta foi extraída do artigoMiniMind: 2 horas de treinamento do zero com a ferramenta de código aberto GPT de parâmetros 26MO
Não pode ser reproduzido sem permissão:Ferramentas de produtividade de IA " Como faço para instalar e verificar o ambiente de tempo de execução do MiniMind em meu próprio computador?
