以下是在Linux中安装配置Llama3的步骤:
curl -fsSL https://ollama.com/install.sh | sh
。ollama run llama3:8b
,可下载并运行80亿参数的Llama3模型。git clone https://github.com/meta-llama/llama3.git
,并进入目录。python3 -m venv venv
,source venv/bin/activate
。pip install -e .
,pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
。llama3/code/utils/init_process_group.py
文件中的nccl为gloo,并设置MASTER_ADDR
、MASTER_PORT
等环境变量。chat.py
文件并添加代码,然后执行python chat.py --ckpt_dir Meta-Llama-3-8B-Instruct/ --tokenizer_path Meta-Llama-3-8B-Instruct/tokenizer.model --max_seq_len 512 --max_batch_size 6
。