ChatGLM HuggingFace调用Bert词向量

开发环境推荐

GPU Dokcer

$ docker pull huggingface/transformers-pytorch-gpu:4.19.4

CPU Dokcer

$ docker pull huggingface/transformers-pytorch-cpu:4.18.0

        我这边使用的是CPU版本,建立容器

$ sudo docker run -it -d -v /Volumes/Yan_Errol/:/workspace --name transformers -p 9000:9000 -p 9001:22 32e2fefc85c5 bash

        进入容器

$ sudo docker exec -it transformers bash

        安装openssh-server

$ apt update && apt install openssh-server
$ mkdir /var/run/sshd && echo 'root:passwd' | chpasswd
$ sed -i 's/PermitRootLogin prohibit-password/PermitRootLogin yes/' /etc/ssh/ss