1 Star 0 Fork 1

diycp2015 / belle

forked from Gitee 极速下载 / belle 
加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
README.md 1.54 KB
一键复制 编辑 原始数据 按行查看 历史
Weiguang Han 提交于 2023-09-26 18:27 . update docker (#522)

docker镜像

国内环境自行替换代理

export https_proxy=...
export http_proxy=...
export all_proxy=...

构建镜像

sudo bash build_dockerfile_upon_transfermers.sh

上传镜像到dockerhub

sudo bash upload_image.sh

下载镜像

已经构建好镜像,无需自行构建

sudo docker pull tothemoon/belle:latest

belle镜像中包含sshd,可以远程连接到容器内部

运行镜像

1. 参考nvidia安装说明安装container-toolkit

2. 自行创建ssh密钥

3. 运行容器

sudo bash docker_run.sh

docker_run.sh文件主要内容如下

docker run --gpus all --ipc=host --ulimit memlock=-1 --ulimit stack=67108864 \
    --network host \
    --privileged \
    [--env https_proxy=$https_proxy \]
    [--env http_proxy=$http_proxy \]
    [--env all_proxy=$all_proxy \]
    --env HF_HOME=$hf_home \
    -it [--rm] \
    --name belle \
    -v $belle_path:$belle_path \
    -v $hf_home:$hf_home \
    -v $ssh_pub_key:/root/.ssh/authorized_keys \
    -w $workdir \
    $docker_user/belle:$tag \
    [--sshd_port 2201 --cmd "echo 'Hello, world!' && /bin/bash"]

[]中内容可忽略

  • --rm:容器退出时销毁,如果长期在容器中工作,可忽略
  • --sshd_port:sshd监听端口,默认是22001
  • --cmd:容器要执行的命令"echo 'Hello, world!' && /bin/bash",可忽略
  • hf_home:huggingface缓存目录
  • $ssh_pub_key:sshd公钥目录
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
C/C++
1
https://gitee.com/diycp2015/belle.git
git@gitee.com:diycp2015/belle.git
diycp2015
belle
belle
main

搜索帮助

344bd9b3 5694891 D2dac590 5694891