ChatGPT 开源实现
基础环境准备
安装显卡驱动
# 查看系统版本
lsb_release -a
# 查看可用驱动版本
ubuntu-drivers devices
# 安装推荐驱动
sudo apt install nvidia-driver-440
# 重启系统
sudo reboot
# 重启后查看驱动
nvidia-smi
# 验证驱动是否安装
cat /proc/driver/nvidia/version
安装 CUDA Toolkit
选择合适的 CUDA Toolkit 版本, 下载地址: CUDA Toolkit
运行安装
# Ubuntu 18.04 系统 # cuda tookit 11.8 下载地址: https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run # cuda tookit 12.1 下载地址: https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run wget download_url sudo 755 download_file sudo sh download_file --tookit --silent --override
如提示 cudnn 动态库找不到, 可参考 NVIDIA cuDNN Documentation 进行安装
加速下载模型
注意
下载模型
wget
git lfs + clone
git lfs install git clone huggingface_model_repo
huggface_hub
from huggingface_hub import snapshot_download # bert-base-chinese 为 hugging_face repo id snapshot_download(repo_id="bert-base-chinese")
基础大模型