synchronization

gpushare-scheduler-extender/deployer/docker/device-plugin-evict/Dockerfile

@@ -0,0 +1,12 @@
+FROM debian:bullseye-slim
+RUN echo \
+    deb [arch=amd64]  http://mirrors.aliyun.com/debian/ bullseye main non-free contrib\
+    > /etc/apt/sources.list
+RUN apt-get update
+RUN apt-get install -y curl tzdata iproute2 bash &&  \
+ 	rm -rf /var/cache/apt/* && \
+ 	cp /usr/share/zoneinfo/Asia/Shanghai /etc/localtime && \
+ 	echo "Asia/Shanghai" >  /etc/timezone && \
+ 	mkdir -p /dp-evict
+ADD dp-evict /dp-evict
+RUN chmod -R +x /dp-evict

gpushare-scheduler-extender/deployer/docker/device-plugin-evict/build.sh

@@ -0,0 +1,16 @@
+#!/usr/bin/env bash
+set -xe
+
+BASEDIR=$(dirname "$0")
+echo "build docker images in $BASEDIR"
+
+TIMESTAMP=$(date +%Y%m%d%H%M)
+
+cd $BASEDIR
+
+# docker build --no-cache -t $IMAGE -f $FILE $BASEDIR
+docker build --network=host -t registry.cn-hangzhou.aliyuncs.com/acs/nvidia-device-plugin-evict:$TIMESTAMP .
+
+docker push registry.cn-hangzhou.aliyuncs.com/acs/nvidia-device-plugin-evict:$TIMESTAMP
+
+echo registry.cn-hangzhou.aliyuncs.com/acs/nvidia-device-plugin-evict:$TIMESTAMP

gpushare-scheduler-extender/deployer/docker/device-plugin-evict/dp-evict/dp-evict-on-host.sh

@@ -0,0 +1,31 @@
+#!/usr/bin/env bash
+
+set -e -x
+
+backup_dir="/etc/kubernetes/manifests-backup"
+
+public::common::log() {
+	echo $(date +"[%Y%m%d %H:%M:%S]: ") $1
+}
+
+public::evict::gpu-device-plugin() {
+	dir=/etc/kubernetes/manifests/
+
+	if [  -f /etc/kubernetes/manifests/nvidia-device-plugin.yml ]; then
+			backup_dir="/etc/kubernetes/manifests-backup/"
+			mkdir -p $backup_dir
+			mv /etc/kubernetes/manifests/nvidia-device-plugin.yml $backup_dir
+		else
+			public::common::log "Skip removing nvidia-device-plugin.yml, because it doesn't exist."
+	fi
+}
+
+main() {
+
+	public::evict::gpu-device-plugin
+
+	touch /ready
+	while sleep 3600; do :; done
+}
+
+main "$@"

gpushare-scheduler-extender/deployer/docker/device-plugin-evict/dp-evict/dp-evict.sh

@@ -0,0 +1,11 @@
+#!/usr/bin/env bash
+set -xe
+
+if [ -d "/k8s-host" ]; then
+	rm -rf /k8s-host/usr/local/dp-evict
+	mkdir -p /k8s-host/usr/local/dp-evict
+	cp -r /dp-evict/* /k8s-host/usr/local/dp-evict
+	chmod -R +x /k8s-host/usr/local/dp-evict
+	chroot /k8s-host /usr/local/dp-evict/dp-evict-on-host.sh "$@"
+	while sleep 3600; do :; done
+fi