YARN-8870. [Submarine] Add submarine installation scripts. (Xun Liu via wangda)
Change-Id: I46e8d9fd32c7745c313030da62da41486a77b3ea
This commit is contained in:
parent
ed08dd3b0c
commit
46d6e00166
|
@ -65,6 +65,14 @@
|
|||
</includes>
|
||||
<fileMode>0755</fileMode>
|
||||
</fileSet>
|
||||
<fileSet>
|
||||
<directory>hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-submarine/installation</directory>
|
||||
<outputDirectory>/share/hadoop/yarn/submarine-installer</outputDirectory>
|
||||
<includes>
|
||||
<include>**/*</include>
|
||||
</includes>
|
||||
<fileMode>0755</fileMode>
|
||||
</fileSet>
|
||||
<fileSet>
|
||||
<directory>hadoop-yarn/conf</directory>
|
||||
<outputDirectory>etc/hadoop</outputDirectory>
|
||||
|
|
|
@ -0,0 +1,74 @@
|
|||
#!/usr/bin/env bash
|
||||
# Licensed to the Apache Software Foundation (ASF) under one
|
||||
# or more contributor license agreements. See the NOTICE file
|
||||
# distributed with this work for additional information
|
||||
# regarding copyright ownership. The ASF licenses this file
|
||||
# to you under the Apache License, Version 2.0 (the
|
||||
# "License"); you may not use this file except in compliance
|
||||
# with the License. You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
##### install config #####
|
||||
|
||||
# DNS
|
||||
LOCAL_DNS_HOST="172.17.0.9" # /etc/resolv.conf
|
||||
YARN_DNS_HOST="10.196.69.173" # yarn dns server ip address
|
||||
|
||||
# etcd hosts list
|
||||
ETCD_HOSTS=(10.196.69.173 10.196.69.174 10.196.69.175)
|
||||
|
||||
# docker registry ip:port
|
||||
DOCKER_REGISTRY="10.120.196.232:5000"
|
||||
|
||||
# Start the http download service on the specified server,
|
||||
# Will download all the dependencies in the http server,
|
||||
# Run the install script on other servers.
|
||||
# Automatically download dependencies from http,
|
||||
# Solve the problem that all servers are slow to download online.
|
||||
# At the same time, you can also manually download the dependencies to the downloads directory.
|
||||
# Offline installation of the system
|
||||
DOWNLOAD_SERVER_IP="10.120.196.236"
|
||||
DOWNLOAD_SERVER_PORT="19000"
|
||||
|
||||
# yarn container-executor config
|
||||
# How to compile container-executor:
|
||||
# Go to the hadoop/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-nodemanager
|
||||
# path and enter the command: mvn package -Pnative -DskipTests
|
||||
# Only the nodemanager is compiled. The path of the compiled container-executor is:
|
||||
# ./target/native/target/usr/local/bin/container-executor
|
||||
YARN_CONTAINER_EXECUTOR_PATH="Please enter the full path to the container-executor here"
|
||||
|
||||
# Keep the same configuration as 'yarn.nodemanager.linux-container-executor.cgroups.hierarchy'
|
||||
# in yarn-site.xml, default '/hadoop-yarn'
|
||||
YARN_HIERARCHY="/hadoop-yarn"
|
||||
|
||||
# Keep the same configuration as 'yarn.nodemanager.local-dirs' in yarn-site.xml
|
||||
YARN_NODEMANAGER_LOCAL_DIRS="/home/hadoop/disk/1/yarn/local,/home/hadoop/disk/2/yarn/local,/home/hadoop/disk/3/yarn/local"
|
||||
|
||||
# Keep the same configuration as 'yarn.nodemanager.log-dirs' in yarn-site.xml
|
||||
YARN_NODEMANAGER_LOG_DIRS="/home/hadoop/disk/1/yarn/logs,/home/hadoop/disk/2/yarn/logs,/home/hadoop/disk/3/yarn/logs"
|
||||
|
||||
##### System component download url address, Generally do not need to be modified #####
|
||||
DOCKER_REPO="https://yum.dockerproject.org/repo/main/centos/7/Packages"
|
||||
DOCKER_ENGINE_RPM="docker-engine-1.12.5-1.el7.centos.x86_64.rpm"
|
||||
DOCKER_ENGINE_SELINUX_RPM="docker-engine-selinux-1.12.5-1.el7.centos.noarch.rpm"
|
||||
|
||||
NVIDIA_DETECT_URL="https://raw.githubusercontent.com/liuxunorg/submarine-installer/packages/nvidia/nvidia-detect"
|
||||
NVIDIA_DOCKER_RPM="nvidia-docker-1.0.1-1.x86_64.rpm"
|
||||
NVIDIA_DOCKER_RPM_URL="https://github.com/NVIDIA/nvidia-docker/releases/download/v1.0.1/${NVIDIA_DOCKER_RPM}"
|
||||
|
||||
ETCD_TAR_GZ="etcd-v3.3.9-linux-amd64.tar.gz"
|
||||
ETCD_DOWNLOAD_URL="https://github.com/etcd-io/etcd/releases/download/v3.3.9/${ETCD_TAR_GZ}"
|
||||
|
||||
CALICO_IPV4POOL_CIDR="192.20.0.0"
|
||||
CALICO_NETWORK_NAME="calico-network"
|
||||
CALICO_DOWNLOAD_URL="https://github.com/projectcalico/cni-plugin/releases/download/v1.11.7/calico"
|
||||
CALICO_IPAM_DOWNLOAD_URL="https://github.com/projectcalico/cni-plugin/releases/download/v1.11.7/calico-ipam"
|
||||
CALICOCTL_DOWNLOAD_URL="https://github.com/projectcalico/calicoctl/releases/download/v3.2.3/calicoctl"
|
|
@ -0,0 +1,116 @@
|
|||
#!/usr/bin/env bash
|
||||
# Licensed to the Apache Software Foundation (ASF) under one
|
||||
# or more contributor license agreements. See the NOTICE file
|
||||
# distributed with this work for additional information
|
||||
# regarding copyright ownership. The ASF licenses this file
|
||||
# to you under the Apache License, Version 2.0 (the
|
||||
# "License"); you may not use this file except in compliance
|
||||
# with the License. You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
# description: sumbarine install scripts.
|
||||
|
||||
ROOT=$(cd "$(dirname "$0")"; pwd)
|
||||
SUBMARINE_INSTALLER_VERSION="v0.7"
|
||||
PACKAGE_DIR=${ROOT}/package
|
||||
SCRIPTS_DIR=${ROOT}/scripts
|
||||
INSTALL_TEMP_DIR=${ROOT}/temp
|
||||
DOWNLOAD_DIR=${ROOT}/downloads
|
||||
DATE=`date +%Y%m%d-%H:%M:%S`
|
||||
LOG=${ROOT}/logs/install.log.`date +%Y%m%d%H%M%S`
|
||||
LOCAL_HOST_IP_LIST=()
|
||||
LOCAL_HOST_IP=""
|
||||
OPERATING_SYSTEM=""
|
||||
DOWNLOAD_HTTP=""
|
||||
|
||||
# import shell script
|
||||
. ${ROOT}/install.conf
|
||||
. ${ROOT}/scripts/calico.sh
|
||||
. ${ROOT}/scripts/docker.sh
|
||||
. ${ROOT}/scripts/download-server.sh
|
||||
. ${ROOT}/scripts/environment.sh
|
||||
. ${ROOT}/scripts/etcd.sh
|
||||
. ${ROOT}/scripts/hadoop.sh
|
||||
. ${ROOT}/scripts/menu.sh
|
||||
. ${ROOT}/scripts/nvidia.sh
|
||||
. ${ROOT}/scripts/nvidia-docker.sh
|
||||
. ${ROOT}/scripts/submarine.sh
|
||||
. ${ROOT}/scripts/utils.sh
|
||||
|
||||
#================================= Main ========================================
|
||||
mkdir $ROOT/logs/ -p
|
||||
mkdir $INSTALL_TEMP_DIR -p
|
||||
mkdir $DOWNLOAD_DIR -p
|
||||
|
||||
source /etc/os-release
|
||||
OPERATING_SYSTEM=$ID
|
||||
|
||||
get_ip_list
|
||||
ipCount=${#LOCAL_HOST_IP_LIST[@]}
|
||||
if [[ $ipCount -eq 1 ]]; then
|
||||
LOCAL_HOST_IP=${LOCAL_HOST_IP_LIST[0]}
|
||||
else
|
||||
echo -e "Detect the network card IP in the server, \e[31m[${LOCAL_HOST_IP_LIST[@]}]\e[0m"
|
||||
echo -n -e "please enter a valid IP address: "
|
||||
|
||||
read ipInput
|
||||
if ! valid_ip $ipInput; then
|
||||
echo -e "you input \e[31m$ipInput\e[0m address format is incorrect! " | tee -a $LOG
|
||||
exit_install
|
||||
else
|
||||
LOCAL_HOST_IP=$ipInput
|
||||
fi
|
||||
fi
|
||||
|
||||
echo -n -e "Please confirm whether the IP address of this machine is \e[31m${LOCAL_HOST_IP}\e[0m?[y|n]"
|
||||
read myselect
|
||||
if [[ "$myselect" != "y" && "$myselect" != "Y" ]]; then
|
||||
exit_install
|
||||
fi
|
||||
|
||||
check_install_conf
|
||||
|
||||
if [[ -n "$DOWNLOAD_SERVER_IP" && -n "$DOWNLOAD_SERVER_PORT" && "$DOWNLOAD_SERVER_IP" != "$LOCAL_HOST_IP" ]]; then
|
||||
DOWNLOAD_HTTP="http://${DOWNLOAD_SERVER_IP}:${DOWNLOAD_SERVER_PORT}"
|
||||
fi
|
||||
|
||||
check_install_user
|
||||
|
||||
# 清理安装临时目录
|
||||
rm $INSTALL_TEMP_DIR/* -rf >>$LOG 2>&1
|
||||
|
||||
menu_index="0"
|
||||
for ((j=1;;j++))
|
||||
do
|
||||
menu
|
||||
case "$menu_index" in
|
||||
"0")
|
||||
menu_index="$menu_choice"
|
||||
;;
|
||||
"1"|"2"|"3"|"4"|"5")
|
||||
# echo "aaaa=$menu_index-$menu_choice"
|
||||
menu_process
|
||||
if [[ $? = 1 ]]; then
|
||||
echo "Press any key to return menu!"
|
||||
read
|
||||
fi
|
||||
;;
|
||||
"a")
|
||||
exit_install
|
||||
;;
|
||||
"q")
|
||||
exit_install
|
||||
;;
|
||||
*)
|
||||
menu_index="0"
|
||||
menu_choice="0"
|
||||
menu
|
||||
;;
|
||||
esac
|
||||
done
|
|
@ -0,0 +1,50 @@
|
|||
# Licensed to the Apache Software Foundation (ASF) under one
|
||||
# or more contributor license agreements. See the NOTICE file
|
||||
# distributed with this work for additional information
|
||||
# regarding copyright ownership. The ASF licenses this file
|
||||
# to you under the Apache License, Version 2.0 (the
|
||||
# "License"); you may not use this file except in compliance
|
||||
# with the License. You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
[Unit]
|
||||
Description=calico node
|
||||
After=docker.service
|
||||
Requires=docker.service
|
||||
|
||||
[Service]
|
||||
User=root
|
||||
PermissionsStartOnly=true
|
||||
ExecStart=/usr/bin/docker run --net=host --privileged --name=calico-node \
|
||||
-e ETCD_ENDPOINTS=ETCD_ENDPOINTS_REPLACE \
|
||||
-e CALICO_LIBNETWORK_ENABLED=true \
|
||||
-e CALICO_NETWORKING_BACKEND=bird \
|
||||
-e CALICO_DISABLE_FILE_LOGGING=true \
|
||||
-e CALICO_IPV4POOL_CIDR=CALICO_IPV4POOL_CIDR_REPLACE/16 \
|
||||
-e CALICO_IPV4POOL_IPIP=always \
|
||||
-e FELIX_DEFAULTENDPOINTTOHOSTACTION=ACCEPT \
|
||||
-e FELIX_IPV6SUPPORT=false \
|
||||
-e FELIX_LOGSEVERITYSCREEN=info \
|
||||
-e FELIX_IPINIPMTU=1440 \
|
||||
-e FELIX_HEALTHENABLED=true \
|
||||
-e IP= \
|
||||
-v /etc/calico/ssl:/etc/calico/ssl \
|
||||
-v /var/run/calico:/var/run/calico \
|
||||
-v /lib/modules:/lib/modules \
|
||||
-v /run/docker/plugins:/run/docker/plugins \
|
||||
-v /var/run/docker.sock:/var/run/docker.sock \
|
||||
-v /var/log/calico:/var/log/calico \
|
||||
calico/node:v2.6.2
|
||||
ExecStop=/usr/bin/docker rm -f calico-node
|
||||
Restart=always
|
||||
RestartSec=10
|
||||
|
||||
[Install]
|
||||
WantedBy=multi-user.target
|
|
@ -0,0 +1,22 @@
|
|||
# Licensed to the Apache Software Foundation (ASF) under one
|
||||
# or more contributor license agreements. See the NOTICE file
|
||||
# distributed with this work for additional information
|
||||
# regarding copyright ownership. The ASF licenses this file
|
||||
# to you under the Apache License, Version 2.0 (the
|
||||
# "License"); you may not use this file except in compliance
|
||||
# with the License. You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
apiVersion: v1
|
||||
kind: calicoApiConfig
|
||||
metadata:
|
||||
spec:
|
||||
datastoreType: "etcdv2"
|
||||
etcdEndpoints: ETCD_ENDPOINTS_REPLACE
|
|
@ -0,0 +1,23 @@
|
|||
# Licensed to the Apache Software Foundation (ASF) under one
|
||||
# or more contributor license agreements. See the NOTICE file
|
||||
# distributed with this work for additional information
|
||||
# regarding copyright ownership. The ASF licenses this file
|
||||
# to you under the Apache License, Version 2.0 (the
|
||||
# "License"); you may not use this file except in compliance
|
||||
# with the License. You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
{
|
||||
"insecure-registries": ["DOCKER_REGISTRY_REPLACE"],
|
||||
"cluster-store":"CLUSTER_STORE_REPLACE",
|
||||
"cluster-advertise":"LOCAL_HOST_IP_REPLACE:2375",
|
||||
"dns": ["YARN_DNS_HOST_REPLACE", "LOCAL_DNS_HOST_REPLACE"],
|
||||
"hosts": ["tcp://LOCAL_HOST_IP_REPLACE:2375","unix:///var/run/docker.sock"]
|
||||
}
|
|
@ -0,0 +1,35 @@
|
|||
# Licensed to the Apache Software Foundation (ASF) under one
|
||||
# or more contributor license agreements. See the NOTICE file
|
||||
# distributed with this work for additional information
|
||||
# regarding copyright ownership. The ASF licenses this file
|
||||
# to you under the Apache License, Version 2.0 (the
|
||||
# "License"); you may not use this file except in compliance
|
||||
# with the License. You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
[Unit]
|
||||
Description=Docker Application Container Engine
|
||||
Documentation=http://docs.docker.io
|
||||
|
||||
[Service]
|
||||
Environment="PATH=/usr/bin:/bin:/sbin:/usr/bin:/usr/sbin"
|
||||
ExecStart=/usr/bin/dockerd --log-level=error
|
||||
ExecStartPost=/sbin/iptables -I FORWARD -s 0.0.0.0/0 -j ACCEPT
|
||||
ExecReload=/bin/kill -s HUP $MAINPID
|
||||
Restart=on-failure
|
||||
RestartSec=5
|
||||
LimitNOFILE=infinity
|
||||
LimitNPROC=infinity
|
||||
LimitCORE=infinity
|
||||
Delegate=yes
|
||||
KillMode=process
|
||||
|
||||
[Install]
|
||||
WantedBy=multi-user.target
|
|
@ -0,0 +1,40 @@
|
|||
# Licensed to the Apache Software Foundation (ASF) under one
|
||||
# or more contributor license agreements. See the NOTICE file
|
||||
# distributed with this work for additional information
|
||||
# regarding copyright ownership. The ASF licenses this file
|
||||
# to you under the Apache License, Version 2.0 (the
|
||||
# "License"); you may not use this file except in compliance
|
||||
# with the License. You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
[Unit]
|
||||
Description=etcd
|
||||
Documentation=https://github.com/coreos/etcd
|
||||
|
||||
[Service]
|
||||
Type=notify
|
||||
Restart=always
|
||||
RestartSec=5s
|
||||
LimitNOFILE=40000
|
||||
TimeoutStartSec=0
|
||||
|
||||
ExecStart=/usr/bin/etcd \
|
||||
--name=ETCD_NODE_NAME_REPLACE \
|
||||
--data-dir=/var/lib/etcd \
|
||||
--listen-client-urls=http://LOCAL_HOST_REPLACE:2379,http://127.0.0.1:2379 \
|
||||
--listen-peer-urls=http://LOCAL_HOST_REPLACE:2380 \
|
||||
--advertise-client-urls=http://LOCAL_HOST_REPLACE:2379 \
|
||||
--initial-advertise-peer-urls=http://LOCAL_HOST_REPLACE:2380 \
|
||||
--initial-cluster=INITIAL_CLUSTER_REPLACE \
|
||||
--initial-cluster-token=etcd-token \
|
||||
--initial-cluster-state=new
|
||||
|
||||
[Install]
|
||||
WantedBy=multi-user.target
|
|
@ -0,0 +1,41 @@
|
|||
# Licensed to the Apache Software Foundation (ASF) under one
|
||||
# or more contributor license agreements. See the NOTICE file
|
||||
# distributed with this work for additional information
|
||||
# regarding copyright ownership. The ASF licenses this file
|
||||
# to you under the Apache License, Version 2.0 (the
|
||||
# "License"); you may not use this file except in compliance
|
||||
# with the License. You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
yarn.nodemanager.local-dirs=YARN_NODEMANAGER_LOCAL_DIRS_REPLACE
|
||||
yarn.nodemanager.linux-container-executor.group=yarn
|
||||
yarn.nodemanager.log-dirs=YARN_NODEMANAGER_LOG_DIRS_REPLACE
|
||||
banned.users=root
|
||||
allowed.system.users=yarn
|
||||
min.user.id=500
|
||||
|
||||
[docker]
|
||||
module.enabled=true
|
||||
docker.binary=/usr/bin/docker
|
||||
docker.allowed.capabilities=SYS_CHROOT,MKNOD,SETFCAP,SETPCAP,FSETID,CHOWN,AUDIT_WRITE,SETGID,NET_RAW,FOWNER,SETUID,DAC_OVERRIDE,KILL,NET_BIND_SERVICE,DAC_READ_SEARCH,SYS_PTRACE,SYS_ADMIN
|
||||
docker.allowed.networks=bridge,host,none,CALICO_NETWORK_NAME_REPLACE
|
||||
docker.allowed.ro-mounts=/etc/group,/etc/passwd,/etc/krb5.conf,YARN_NODEMANAGER_LOCAL_DIRS_REPLACE,regex:^nvidia_driver_.*$
|
||||
docker.allowed.rw-mounts=YARN_NODEMANAGER_LOCAL_DIRS_REPLACE,YARN_NODEMANAGER_LOG_DIRS_REPLACE
|
||||
docker.privileged-containers.enabled=false
|
||||
docker.trusted.registries=local,centos,hortonworks,DOCKER_REGISTRY_REPLACE
|
||||
docker.allowed.volume-drivers=nvidia-docker
|
||||
docker.allowed.devices=regex:^/dev/nvidia.*$
|
||||
|
||||
[gpu]
|
||||
module.enabled=true
|
||||
|
||||
[cgroups]
|
||||
root=/sys/fs/cgroup
|
||||
yarn-hierarchy=YARN_HIERARCHY_REPLACE
|
|
@ -0,0 +1,25 @@
|
|||
#!/usr/bin/env bash
|
||||
# Licensed to the Apache Software Foundation (ASF) under one
|
||||
# or more contributor license agreements. See the NOTICE file
|
||||
# distributed with this work for additional information
|
||||
# regarding copyright ownership. The ASF licenses this file
|
||||
# to you under the Apache License, Version 2.0 (the
|
||||
# "License"); you may not use this file except in compliance
|
||||
# with the License. You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
# Solution: Grant user yarn the access to /sys/fs/cgroup/cpu,cpuacct,
|
||||
# which is the subfolder of cgroup mount destination.
|
||||
chown :yarn -R /sys/fs/cgroup/cpu,cpuacct
|
||||
chmod g+rwx -R /sys/fs/cgroup/cpu,cpuacct
|
||||
|
||||
# If GPUs are used,the access to cgroup devices folder is neede as well
|
||||
chown :yarn -R /sys/fs/cgroup/devices
|
||||
chmod g+rwx -R /sys/fs/cgroup/devices
|
|
@ -0,0 +1,224 @@
|
|||
#!/usr/bin/env bash
|
||||
# Licensed to the Apache Software Foundation (ASF) under one
|
||||
# or more contributor license agreements. See the NOTICE file
|
||||
# distributed with this work for additional information
|
||||
# regarding copyright ownership. The ASF licenses this file
|
||||
# to you under the Apache License, Version 2.0 (the
|
||||
# "License"); you may not use this file except in compliance
|
||||
# with the License. You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
## @description download calico bin
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function download_calico_bin()
|
||||
{
|
||||
# submarin http server
|
||||
if [[ -n "$DOWNLOAD_HTTP" ]]; then
|
||||
MY_CALICOCTL_DOWNLOAD_URL=${DOWNLOAD_HTTP}/downloads/calico/calicoctl
|
||||
MY_CALICO_DOWNLOAD_URL=${DOWNLOAD_HTTP}/downloads/calico/calico
|
||||
MY_CALICO_IPAM_DOWNLOAD_URL=${DOWNLOAD_HTTP}/downloads/calico/calico-ipam
|
||||
else
|
||||
MY_CALICOCTL_DOWNLOAD_URL=${CALICOCTL_DOWNLOAD_URL}
|
||||
MY_CALICO_DOWNLOAD_URL=${CALICO_DOWNLOAD_URL}
|
||||
MY_CALICO_IPAM_DOWNLOAD_URL=${CALICO_IPAM_DOWNLOAD_URL}
|
||||
fi
|
||||
|
||||
mkdir -p ${DOWNLOAD_DIR}/calico
|
||||
|
||||
if [[ -f ${DOWNLOAD_DIR}/calico/calico ]]; then
|
||||
echo "${DOWNLOAD_DIR}/calico/calico is exist."
|
||||
else
|
||||
echo "download ${MY_CALICO_DOWNLOAD_URL} ..."
|
||||
wget -P ${DOWNLOAD_DIR}/calico ${MY_CALICO_DOWNLOAD_URL}
|
||||
fi
|
||||
|
||||
if [[ -f ${DOWNLOAD_DIR}/calico/calicoctl ]]; then
|
||||
echo "${DOWNLOAD_DIR}/calico is exist."
|
||||
else
|
||||
echo "download ${MY_CALICOCTL_DOWNLOAD_URL} ..."
|
||||
wget -P ${DOWNLOAD_DIR}/calico ${MY_CALICOCTL_DOWNLOAD_URL}
|
||||
fi
|
||||
|
||||
if [[ -f ${DOWNLOAD_DIR}/calico/calico-ipam ]]; then
|
||||
echo "${DOWNLOAD_DIR}/calico/calico-ipam is exist."
|
||||
else
|
||||
echo "download ${MY_CALICO_IPAM_DOWNLOAD_URL} ..."
|
||||
wget -P ${DOWNLOAD_DIR}/calico ${MY_CALICO_IPAM_DOWNLOAD_URL}
|
||||
fi
|
||||
}
|
||||
|
||||
## @description install calico bin
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function install_calico_bin()
|
||||
{
|
||||
download_calico_bin
|
||||
|
||||
cp -f ${DOWNLOAD_DIR}/calico/calico /usr/bin/calico
|
||||
cp -f ${DOWNLOAD_DIR}/calico/calicoctl /usr/bin/calicoctl
|
||||
cp -f ${DOWNLOAD_DIR}/calico/calico-ipam /usr/bin/calico-ipam
|
||||
|
||||
chmod +x /usr/bin/calico
|
||||
chmod +x /usr/bin/calicoctl
|
||||
chmod +x /usr/bin/calico-ipam
|
||||
}
|
||||
|
||||
## @description install calico config
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function install_calico_config()
|
||||
{
|
||||
mkdir -p /etc/calico
|
||||
|
||||
cp -rf ${PACKAGE_DIR}/calico ${INSTALL_TEMP_DIR}/
|
||||
|
||||
# 1. replace etcdEndpoints
|
||||
# etcdEndpoints: https://10.196.69.173:2379,https://10.196.69.174:2379,https://10.196.69.175:2379
|
||||
etcdEndpoints=''
|
||||
index=0
|
||||
etcdHostsSize=${#ETCD_HOSTS[@]}
|
||||
for item in ${ETCD_HOSTS[@]}
|
||||
do
|
||||
index=$(($index+1))
|
||||
etcdEndpoints="${etcdEndpoints}http:\/\/${item}:2379"
|
||||
if [[ ${index} -lt ${etcdHostsSize} ]]; then
|
||||
etcdEndpoints=${etcdEndpoints}","
|
||||
fi
|
||||
done
|
||||
# echo "etcdEndpoints=${etcdEndpoints}"
|
||||
sed -i "s/ETCD_ENDPOINTS_REPLACE/${etcdEndpoints}/g" $INSTALL_TEMP_DIR/calico/calicoctl.cfg >>$LOG
|
||||
|
||||
if [[ ! -d /etc/calico ]]; then
|
||||
mkdir /etc/calico
|
||||
else
|
||||
rm -rf /etc/calico/*
|
||||
fi
|
||||
|
||||
cp -f $INSTALL_TEMP_DIR/calico/calicoctl.cfg /etc/calico/calicoctl.cfg
|
||||
|
||||
sed -i "s/ETCD_ENDPOINTS_REPLACE/${etcdEndpoints}/g" $INSTALL_TEMP_DIR/calico/calico-node.service >>$LOG
|
||||
sed -i "s/CALICO_IPV4POOL_CIDR_REPLACE/${CALICO_IPV4POOL_CIDR}/g" $INSTALL_TEMP_DIR/calico/calico-node.service >>$LOG
|
||||
cp $INSTALL_TEMP_DIR/calico/calico-node.service /etc/systemd/system/ >>$LOG
|
||||
|
||||
systemctl daemon-reload
|
||||
systemctl enable calico-node.service
|
||||
}
|
||||
|
||||
## @description modify kernel network config
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function kernel_network_config()
|
||||
{
|
||||
if [ `grep -c "net.ipv4.conf.all.rp_filter=1" /etc/sysctl.conf` -eq '0' ]; then
|
||||
echo "net.ipv4.conf.all.rp_filter=1" >>/etc/sysctl.conf
|
||||
fi
|
||||
|
||||
if [ `grep -c "net.ipv4.ip_forward=1" /etc/sysctl.conf` -eq '0' ]; then
|
||||
echo "net.ipv4.ip_forward=1" >>/etc/sysctl.conf
|
||||
fi
|
||||
|
||||
sysctl -p
|
||||
}
|
||||
|
||||
## @description check if the calico-network exist
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function calico_network_exist()
|
||||
{
|
||||
local dockerNetwokInfo=`docker network ls --filter NAME=${CALICO_NETWORK_NAME}`
|
||||
echo ${dockerNetwokInfo} | grep ${CALICO_NETWORK_NAME}
|
||||
}
|
||||
|
||||
## @description verification calico
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function verification_calico()
|
||||
{
|
||||
echo " ===== Check if the network between 2 containers can be connected ====="
|
||||
local claicoNetworkExist=`calico_network_exist`
|
||||
if [[ "$claicoNetworkExist" = "" ]]; then
|
||||
echo "Create a calico network"
|
||||
docker network create --driver calico --ipam-driver calico-ipam ${CALICO_NETWORK_NAME}
|
||||
else
|
||||
echo "calico network ${CALICO_NETWORK_NAME} is exist."
|
||||
fi
|
||||
|
||||
local verifyA="verify-calico-network-A"
|
||||
local verifyAInfo=`containers_exist ${verifyA}`
|
||||
if [[ -n "$verifyAInfo" ]]; then
|
||||
echo "Delete existing container ${verifyA}."
|
||||
docker stop ${verifyA}
|
||||
docker rm ${verifyA}
|
||||
fi
|
||||
echo "Create containers verify-calico-network-A"
|
||||
docker run --net ${CALICO_NETWORK_NAME} --name ${verifyA} -tid busybox
|
||||
|
||||
local verifyB="verify-calico-network-B"
|
||||
local verifyBInfo=`containers_exist ${verifyB}`
|
||||
if [[ -n "$verifyBInfo" ]]; then
|
||||
echo "Delete existing container ${verifyB}."
|
||||
docker stop ${verifyB}
|
||||
docker rm ${verifyB}
|
||||
fi
|
||||
echo "Create containers verify-calico-network-B"
|
||||
docker run --net ${CALICO_NETWORK_NAME} --name ${verifyB} -tid busybox
|
||||
|
||||
echo -e "\033[33m${verifyA} ping ${verifyB}\033[0m"
|
||||
docker exec ${verifyA} ping ${verifyB} -c 5
|
||||
}
|
||||
|
||||
## @description install calico
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function install_calico()
|
||||
{
|
||||
kernel_network_config
|
||||
install_calico_bin
|
||||
install_calico_config
|
||||
start_calico
|
||||
verification_calico
|
||||
}
|
||||
|
||||
## @description uninstall calico
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function uninstall_calico()
|
||||
{
|
||||
echo "stop calico-node.service"
|
||||
systemctl stop calico-node.service
|
||||
|
||||
echo "rm /usr/bin/calico ..."
|
||||
rm /usr/bin/calicoctl
|
||||
rm /usr/bin/calico
|
||||
rm /usr/bin/calico-ipam
|
||||
|
||||
rm -rf /etc/calico/
|
||||
rm /etc/systemd/system/calico-node.service
|
||||
systemctl daemon-reload
|
||||
}
|
||||
|
||||
## @description start calico
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function start_calico()
|
||||
{
|
||||
systemctl restart calico-node.service
|
||||
systemctl status calico-node.service
|
||||
}
|
||||
|
||||
## @description stop calico
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function stop_calico()
|
||||
{
|
||||
systemctl stop calico-node.service
|
||||
systemctl status calico-node.service
|
||||
}
|
|
@ -0,0 +1,166 @@
|
|||
#!/usr/bin/env bash
|
||||
# Licensed to the Apache Software Foundation (ASF) under one
|
||||
# or more contributor license agreements. See the NOTICE file
|
||||
# distributed with this work for additional information
|
||||
# regarding copyright ownership. The ASF licenses this file
|
||||
# to you under the Apache License, Version 2.0 (the
|
||||
# "License"); you may not use this file except in compliance
|
||||
# with the License. You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
## @description download docker rmp
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function download_docker_rpm()
|
||||
{
|
||||
# download http server
|
||||
if [[ -n "$DOWNLOAD_HTTP" ]]; then
|
||||
MY_DOCKER_ENGINE_SELINUX_RPM="${DOWNLOAD_HTTP}/downloads/docker/${DOCKER_ENGINE_SELINUX_RPM}"
|
||||
MY_DOCKER_ENGINE_RPM="${DOWNLOAD_HTTP}/downloads/docker/${DOCKER_ENGINE_RPM}"
|
||||
else
|
||||
MY_DOCKER_ENGINE_SELINUX_RPM=${DOCKER_REPO}/${DOCKER_ENGINE_SELINUX_RPM}
|
||||
MY_DOCKER_ENGINE_RPM=${DOCKER_REPO}/${DOCKER_ENGINE_RPM}
|
||||
fi
|
||||
|
||||
# download docker rpm
|
||||
if [[ -f ${DOWNLOAD_DIR}/docker/${DOCKER_ENGINE_SELINUX_RPM} ]]; then
|
||||
echo "${DOWNLOAD_DIR}/docker/${DOCKER_ENGINE_SELINUX_RPM} is exist."
|
||||
else
|
||||
echo "download ${MY_DOCKER_ENGINE_SELINUX_RPM} ..."
|
||||
wget -P ${DOWNLOAD_DIR}/docker/ ${MY_DOCKER_ENGINE_SELINUX_RPM}
|
||||
fi
|
||||
|
||||
if [[ -f ${DOWNLOAD_DIR}/docker/${DOCKER_ENGINE_RPM} ]]; then
|
||||
echo "${DOWNLOAD_DIR}/docker/${DOCKER_ENGINE_RPM} is exist."
|
||||
else
|
||||
echo "download ${MY_DOCKER_ENGINE_RPM} ..."
|
||||
wget -P ${DOWNLOAD_DIR}/docker/ ${MY_DOCKER_ENGINE_RPM}
|
||||
fi
|
||||
}
|
||||
|
||||
## @description install docker bin
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function install_docker_bin()
|
||||
{
|
||||
download_docker_rpm
|
||||
|
||||
yum -y localinstall ${DOWNLOAD_DIR}/docker/${DOCKER_ENGINE_SELINUX_RPM}
|
||||
yum -y localinstall ${DOWNLOAD_DIR}/docker/${DOCKER_ENGINE_RPM}
|
||||
}
|
||||
|
||||
## @description uninstall docker bin
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function uninstall_docker_bin()
|
||||
{
|
||||
download_docker_rpm
|
||||
|
||||
yum -y remove ${DOWNLOAD_DIR}/docker/${DOCKER_ENGINE_SELINUX_RPM}
|
||||
yum -y remove ${DOWNLOAD_DIR}/docker/${DOCKER_ENGINE_RPM}
|
||||
}
|
||||
|
||||
## @description install docker config
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function install_docker_config()
|
||||
{
|
||||
rm -rf ${INSTALL_TEMP_DIR}/docker
|
||||
cp -rf ${PACKAGE_DIR}/docker ${INSTALL_TEMP_DIR}/
|
||||
|
||||
# replace cluster-store
|
||||
# "cluster-store":"etcd://10.196.69.173:2379,10.196.69.174:2379,10.196.69.175:2379"
|
||||
# char '/' need to escape '\/'
|
||||
clusterStore="etcd:\/\/"
|
||||
index=1
|
||||
etcdHostsSize=${#ETCD_HOSTS[@]}
|
||||
for item in ${ETCD_HOSTS[@]}
|
||||
do
|
||||
clusterStore="${clusterStore}${item}:2379"
|
||||
if [[ ${index} -lt ${etcdHostsSize}-1 ]]; then
|
||||
clusterStore=${clusterStore}","
|
||||
fi
|
||||
index=$(($index+1))
|
||||
done
|
||||
echo "clusterStore=${clusterStore}"
|
||||
sed -i "s/CLUSTER_STORE_REPLACE/${clusterStore}/g" $INSTALL_TEMP_DIR/docker/daemon.json >>$LOG
|
||||
|
||||
sed -i "s/DOCKER_REGISTRY_REPLACE/${DOCKER_REGISTRY}/g" $INSTALL_TEMP_DIR/docker/daemon.json >>$LOG
|
||||
sed -i "s/LOCAL_HOST_IP_REPLACE/${LOCAL_HOST_IP}/g" $INSTALL_TEMP_DIR/docker/daemon.json >>$LOG
|
||||
sed -i "s/YARN_DNS_HOST_REPLACE/${YARN_DNS_HOST}/g" $INSTALL_TEMP_DIR/docker/daemon.json >>$LOG
|
||||
sed -i "s/LOCAL_DNS_HOST_REPLACE/${LOCAL_DNS_HOST}/g" $INSTALL_TEMP_DIR/docker/daemon.json >>$LOG
|
||||
|
||||
# Delete the ASF license comment in the daemon.json file, otherwise it will cause a json format error.
|
||||
sed -i '1,16d' $INSTALL_TEMP_DIR/docker/daemon.json
|
||||
|
||||
if [ ! -d "/etc/docker" ]; then
|
||||
mkdir /etc/docker
|
||||
fi
|
||||
|
||||
cp $INSTALL_TEMP_DIR/docker/daemon.json /etc/docker/
|
||||
cp $INSTALL_TEMP_DIR/docker/docker.service /etc/systemd/system/ >>$LOG
|
||||
}
|
||||
|
||||
## @description install docker
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function install_docker()
|
||||
{
|
||||
install_docker_bin
|
||||
install_docker_config
|
||||
|
||||
systemctl daemon-reload
|
||||
systemctl enable docker.service
|
||||
}
|
||||
|
||||
## @description unstall docker
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function uninstall_docker()
|
||||
{
|
||||
echo "stop docker service"
|
||||
systemctl stop docker
|
||||
|
||||
echo "remove docker"
|
||||
uninstall_docker_bin
|
||||
|
||||
rm /etc/docker/daemon.json >>$LOG
|
||||
rm /etc/systemd/system/docker.service >>$LOG
|
||||
|
||||
systemctl daemon-reload
|
||||
}
|
||||
|
||||
## @description start docker
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function start_docker()
|
||||
{
|
||||
systemctl restart docker
|
||||
systemctl status docker
|
||||
docker info
|
||||
}
|
||||
|
||||
## @description stop docker
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function stop_docker()
|
||||
{
|
||||
systemctl stop docker
|
||||
systemctl status docker
|
||||
}
|
||||
|
||||
## @description check if the containers exist
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function containers_exist()
|
||||
{
|
||||
local dockerContainersInfo=`docker ps -a --filter NAME=$1`
|
||||
echo ${dockerContainersInfo} | grep $1
|
||||
}
|
|
@ -0,0 +1,42 @@
|
|||
#!/usr/bin/env bash
|
||||
# Licensed to the Apache Software Foundation (ASF) under one
|
||||
# or more contributor license agreements. See the NOTICE file
|
||||
# distributed with this work for additional information
|
||||
# regarding copyright ownership. The ASF licenses this file
|
||||
# to you under the Apache License, Version 2.0 (the
|
||||
# "License"); you may not use this file except in compliance
|
||||
# with the License. You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
## @description start download server
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function start_download_server()
|
||||
{
|
||||
if [[ "$DOWNLOAD_SERVER_IP" != "$LOCAL_HOST_IP" ]]; then
|
||||
echo -e "\033[31mERROR: Only $DOWNLOAD_SERVER_IP can start the download service.\033[0m"
|
||||
return 1
|
||||
fi
|
||||
|
||||
echo -e "You can put the install package file in the \033[34m${DOWNLOAD_DIR}\033[0m folder first, Or automatic download."
|
||||
echo -n "Do you want to start download http server?[y|n]"
|
||||
read myselect
|
||||
if [[ "$myselect" = "y" || "$myselect" = "Y" ]]
|
||||
then
|
||||
download_etcd_bin
|
||||
download_calico_bin
|
||||
download_docker_rpm
|
||||
download_nvidia_driver
|
||||
download_nvidia_docker_bin
|
||||
download_yarn_container_executor
|
||||
|
||||
python -m SimpleHTTPServer ${DOWNLOAD_SERVER_PORT}
|
||||
fi
|
||||
}
|
|
@ -0,0 +1,213 @@
|
|||
#!/usr/bin/env bash
|
||||
# Licensed to the Apache Software Foundation (ASF) under one
|
||||
# or more contributor license agreements. See the NOTICE file
|
||||
# distributed with this work for additional information
|
||||
# regarding copyright ownership. The ASF licenses this file
|
||||
# to you under the Apache License, Version 2.0 (the
|
||||
# "License"); you may not use this file except in compliance
|
||||
# with the License. You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
## @description check operation System
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function check_operationSystem()
|
||||
{
|
||||
echo -e "The submarine assembly support \033[32m[centos-release-7-3.1611.el7.centos.x86_64]\033[0m or higher operating system version."
|
||||
|
||||
case ${OPERATING_SYSTEM} in
|
||||
centos)
|
||||
local operationSystemVersion=`rpm --query centos-release`
|
||||
echo -e "The current operating system version is \e[31m[${operationSystemVersion}]\e[0m" | tee -a $LOG
|
||||
;;
|
||||
*)
|
||||
echo -e "\033[31mWARN: The submarine assembly Unsupported [${OPERATING_SYSTEM}] operating system\033[0m"
|
||||
;;
|
||||
esac
|
||||
}
|
||||
|
||||
## @description update operation System Kernel
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function update_operationSystemKernel()
|
||||
{
|
||||
echo "If the server is unable to connect to the network, execute the following command yourself:
|
||||
wget http://vault.centos.org/7.3.1611/os/x86_64/Packages/kernel-headers-3.10.0-514.el7.x86_64.rpm
|
||||
rpm -ivh kernel-headers-3.10.0-514.el7.x86_64.rpm"
|
||||
|
||||
echo -n "Do you want to kernel upgrades?[y|n]"
|
||||
read myselect
|
||||
if [[ "$myselect" = "y" || "$myselect" = "Y" ]]
|
||||
then
|
||||
echo "Now try to use the yum command for kernel upgrades ..."
|
||||
yum install kernel-devel-$(uname -r) kernel-headers-$(uname -r)
|
||||
|
||||
local kernelVersion=`uname -r`
|
||||
echo -e "After the upgrade, the operating system kernel version is \e[31m${kernelVersion}\e[0m" | tee -a $LOG
|
||||
fi
|
||||
}
|
||||
|
||||
## @description check operation system kernel
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function check_operationSystemKernel()
|
||||
{
|
||||
case ${OPERATING_SYSTEM} in
|
||||
centos)
|
||||
local kernelVersion=`uname -r`
|
||||
|
||||
echo -e "Submarine support operating system kernel version is \033[32m 3.10.0-514.el7.x86_64 \033[0m" | tee -a $LOG
|
||||
echo -e "Current operating system kernel version is \e[31m${kernelVersion}\e[0m" | tee -a $LOG
|
||||
|
||||
update_operationSystemKernel
|
||||
;;
|
||||
*)
|
||||
echo -e "\033[31m WARN: The submarine assembly Unsupported operating system [${OPERATING_SYSTEM}] \033[0m"
|
||||
;;
|
||||
esac
|
||||
}
|
||||
|
||||
## @description get gcc version
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function get_gcc_version()
|
||||
{
|
||||
local gccVersion=`gcc --version`
|
||||
version=${gccVersion%Copyright*}
|
||||
echo $version
|
||||
}
|
||||
|
||||
## @description install gcc
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function install_gcc()
|
||||
{
|
||||
echo -n "Do you want to install gcc?[y|n]"
|
||||
read myselect
|
||||
if [[ "$myselect" = "y" || "$myselect" = "Y" ]]; then
|
||||
echo "Execute the yum install gcc make g++ command"
|
||||
yum install gcc make g++
|
||||
|
||||
local gccVersion=`gcc --version`
|
||||
echo -e "After the install, the gcc version is \e[31m${gccVersion}\e[0m" | tee -a $LOG
|
||||
fi
|
||||
}
|
||||
|
||||
## @description check gcc Version
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function check_gccVersion()
|
||||
{
|
||||
local gccVersionInfo=`gcc --version`
|
||||
local gccVersion=${gccVersionInfo%Copyright*}
|
||||
|
||||
if [[ "$gccVersion" = "" ]]; then
|
||||
echo "The gcc was not installed on the system. Automated installation ..."
|
||||
install_gcc
|
||||
else
|
||||
echo -e "Submarine gcc version need \033[34mgcc (GCC) 4.8.5 20150623 (Red Hat 4.8.5-11)\033[0m or higher."
|
||||
echo -e "Current gcc version was \033[34m${gccVersion}\033[0m"
|
||||
fi
|
||||
}
|
||||
|
||||
## @description check GPU
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function check_GPU()
|
||||
{
|
||||
gpuInfo=`lspci | grep -i nvidia`
|
||||
|
||||
if [[ "$gpuInfo" = "" ]]; then
|
||||
echo -e "\033[31mERROR: The system did not detect the GPU graphics card.\033[0m"
|
||||
else
|
||||
echo -e "\033[32mINFO: The system detect the GPU graphics card.\033[0m"
|
||||
fi
|
||||
}
|
||||
|
||||
## @description check user group
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function check_userGroup()
|
||||
{
|
||||
echo -e "check hadoop user group ..."
|
||||
|
||||
echo -e "Hadoop runs the required user [hdfs, mapred, yarn] and groups [hdfs, mapred, yarn, hadoop] installed by ambari."
|
||||
echo -e "If you are not using ambari for hadoop installation,
|
||||
then you can add the user and group by root by executing the following command:
|
||||
\033[34madduser hdfs
|
||||
adduser mapred
|
||||
adduser yarn
|
||||
addgroup hadoop
|
||||
usermod -aG hdfs,hadoop hdfs
|
||||
usermod -aG mapred,hadoop mapred
|
||||
usermod -aG yarn,hadoop yarn
|
||||
usermod -aG hdfs,hadoop hadoop
|
||||
groupadd docker
|
||||
usermod -aG docker yarn
|
||||
usermod -aG docker hadoop\033[0m\n"
|
||||
|
||||
echo -e "check docker user group ..."
|
||||
# check user group
|
||||
DOCKER_USER_GROUP='docker'
|
||||
egrep "^${DOCKER_USER_GROUP}" /etc/group >& /dev/null
|
||||
if [[ $? -ne 0 ]]; then
|
||||
echo -e "user group ${DOCKER_USER_GROUP} does not exist, Please execute the following command:"
|
||||
echo -e "\033[34mgroupadd $DOCKER_USER_GROUP\033[0m"
|
||||
fi
|
||||
|
||||
# check user
|
||||
USER_GROUP=(yarn hadoop)
|
||||
for user in ${USER_GROUP[@]}
|
||||
do
|
||||
egrep "^${user}" /etc/passwd >& /dev/null
|
||||
if [[ $? -ne 0 ]]; then
|
||||
echo -e "User ${user} does not exist, Please execute the following command:"
|
||||
echo -e "\033[34madduser ${user}\033[0m"
|
||||
echo -e "\033[34musermod -aG ${DOCKER_USER_GROUP} ${user}\033[0m"
|
||||
fi
|
||||
|
||||
echo -e "Please execute the following command:"
|
||||
echo -e "\033[34musermod -aG ${DOCKER_USER_GROUP} ${user}\033[0m"
|
||||
done
|
||||
}
|
||||
|
||||
## @description Some preparatory work for nvidia driver installation
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function prepare_nvidia_environment()
|
||||
{
|
||||
echo "prepare nvidia environment ..."
|
||||
|
||||
yum -y update
|
||||
yum install kernel-devel-$(uname -r) kernel-headers-$(uname -r)
|
||||
|
||||
yum -y install epel-release
|
||||
yum -y install dkms
|
||||
|
||||
echo -e "\033[34m ===== Please manually execute the following command =====
|
||||
# 1. Disable nouveau
|
||||
# Add the content 'rd.driver.blacklist=nouveau nouveau.modeset=0'
|
||||
# to the 'GRUB_CMDLINE_LINUX' configuration item in the /etc/default/grub file.
|
||||
root:> vi /etc/default/grub
|
||||
vi:> GRUB_CMDLINE_LINUX=\"rd.driver.blacklist=nouveau nouveau.modeset=0 ...\"
|
||||
|
||||
# 2. Generate configuration
|
||||
root:> grub2-mkconfig -o /boot/grub2/grub.cfg
|
||||
|
||||
# 3. Open (new) /etc/modprobe.d/blacklist.conf, add content 'blacklist nouveau'
|
||||
root:> vi /etc/modprobe.d/blacklist.conf
|
||||
vi:> blacklist nouveau
|
||||
|
||||
# 4. Update configuration and reboot
|
||||
root:> mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r)-nouveau.img
|
||||
root:> dracut /boot/initramfs-$(uname -r).img $(uname -r)
|
||||
root:> reboot
|
||||
\033[0m"
|
||||
}
|
|
@ -0,0 +1,152 @@
|
|||
#!/usr/bin/env bash
|
||||
# Licensed to the Apache Software Foundation (ASF) under one
|
||||
# or more contributor license agreements. See the NOTICE file
|
||||
# distributed with this work for additional information
|
||||
# regarding copyright ownership. The ASF licenses this file
|
||||
# to you under the Apache License, Version 2.0 (the
|
||||
# "License"); you may not use this file except in compliance
|
||||
# with the License. You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
## @description download etcd bin
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function download_etcd_bin()
|
||||
{
|
||||
# my download http server
|
||||
if [[ -n "$DOWNLOAD_HTTP" ]]; then
|
||||
MY_ETCD_DOWNLOAD_URL="${DOWNLOAD_HTTP}/downloads/etcd/${ETCD_TAR_GZ}"
|
||||
else
|
||||
MY_ETCD_DOWNLOAD_URL=${ETCD_DOWNLOAD_URL}
|
||||
fi
|
||||
|
||||
if [[ -f "${DOWNLOAD_DIR}/etcd/${ETCD_TAR_GZ}" ]]; then
|
||||
echo "${DOWNLOAD_DIR}/etcd/${ETCD_TAR_GZ} is exist."
|
||||
else
|
||||
echo "download ${MY_ETCD_DOWNLOAD_URL} ..."
|
||||
wget -P ${DOWNLOAD_DIR}/etcd ${MY_ETCD_DOWNLOAD_URL}
|
||||
fi
|
||||
}
|
||||
|
||||
## @description install etcd bin
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function install_etcd_bin()
|
||||
{
|
||||
download_etcd_bin
|
||||
|
||||
# install etcd bin
|
||||
mkdir -p ${INSTALL_TEMP_DIR}
|
||||
rm -rf ${INSTALL_TEMP_DIR}/etcd-*-linux-amd6
|
||||
tar zxvf ${DOWNLOAD_DIR}/etcd/${ETCD_TAR_GZ} -C ${INSTALL_TEMP_DIR}
|
||||
|
||||
cp -f ${INSTALL_TEMP_DIR}/etcd-*-linux-amd64/etcd /usr/bin
|
||||
cp -f ${INSTALL_TEMP_DIR}/etcd-*-linux-amd64/etcdctl /usr/bin
|
||||
|
||||
mkdir -p /var/lib/etcd
|
||||
chmod -R a+rw /var/lib/etcd
|
||||
}
|
||||
|
||||
## @description install etcd config
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function install_etcd_config()
|
||||
{
|
||||
# config etcd.service
|
||||
rm -rf ${INSTALL_TEMP_DIR}/etcd
|
||||
cp -rf ${PACKAGE_DIR}/etcd ${INSTALL_TEMP_DIR}/
|
||||
|
||||
# 1. Replace name with ETCD_NODE_NAME_REPLACE based on the location of the local IP in $ETCD_HOSTS
|
||||
indexEtcdList=$(indexByEtcdHosts ${LOCAL_HOST_IP})
|
||||
# echo ${indexEtcdList}
|
||||
etcdNodeName="etcdnode${indexEtcdList}"
|
||||
# echo ${etcdNodeName}
|
||||
sed -i "s/ETCD_NODE_NAME_REPLACE/${etcdNodeName}/g" $INSTALL_TEMP_DIR/etcd/etcd.service >>$LOG
|
||||
|
||||
# 2. Replace local IP address
|
||||
sed -i "s/LOCAL_HOST_REPLACE/${LOCAL_HOST_IP}/g" $INSTALL_TEMP_DIR/etcd/etcd.service >>$LOG
|
||||
|
||||
# 3. Replace the initial-cluster parameter
|
||||
# --initial-cluster=etcdnode1=http://10.196.69.173:2380,etcdnode2=http://10.196.69.174:2380,etcdnode3=http://10.196.69.175:2380 \
|
||||
initialCluster=''
|
||||
index=0
|
||||
etcdHostsSize=${#ETCD_HOSTS[@]}
|
||||
for item in ${ETCD_HOSTS[@]}
|
||||
do
|
||||
# char '/' need to escape '\/'
|
||||
initialCluster="${initialCluster}etcdnode${index}=http:\/\/${item}:2380"
|
||||
if [[ ${index} -lt ${etcdHostsSize}-1 ]]; then
|
||||
initialCluster=${initialCluster}","
|
||||
fi
|
||||
index=$(($index+1))
|
||||
done
|
||||
#echo "initialCluster=${initialCluster}"
|
||||
sed -i "s/INITIAL_CLUSTER_REPLACE/${initialCluster}/g" $INSTALL_TEMP_DIR/etcd/etcd.service >>$LOG
|
||||
|
||||
cp $INSTALL_TEMP_DIR/etcd/etcd.service /etc/systemd/system/ >>$LOG
|
||||
}
|
||||
|
||||
## @description install etcd
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function install_etcd()
|
||||
{
|
||||
index=$(indexByEtcdHosts ${LOCAL_HOST_IP})
|
||||
if [ -z "$index" ]; then
|
||||
echo -e "STOP: This host\033[31m[${LOCAL_HOST_IP}]\033[0m is not in the ETCD server list\033[31m[${ETCD_HOSTS[@]}]\033[0m"
|
||||
return 1
|
||||
fi
|
||||
|
||||
install_etcd_bin
|
||||
|
||||
install_etcd_config
|
||||
|
||||
systemctl daemon-reload
|
||||
systemctl enable etcd.service
|
||||
}
|
||||
|
||||
## @description uninstall etcd
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function uninstall_etcd()
|
||||
{
|
||||
echo "stop etcd.service"
|
||||
systemctl stop etcd.service
|
||||
|
||||
echo "rm etcd ..."
|
||||
rm /usr/bin/etcd
|
||||
rm /usr/bin/etcdctl
|
||||
rm -rf /var/lib/etcd
|
||||
rm /etc/systemd/system/etcd.service
|
||||
|
||||
systemctl daemon-reload
|
||||
}
|
||||
|
||||
## @description start etcd
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function start_etcd()
|
||||
{
|
||||
systemctl restart etcd.service
|
||||
|
||||
echo " ===== Check the status of the etcd service ====="
|
||||
echo " exec etcdctl cluster-health"
|
||||
etcdctl cluster-health
|
||||
echo " exec etcdctl cluster-health"
|
||||
etcdctl member list
|
||||
}
|
||||
|
||||
## @description stop etcd
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function stop_etcd()
|
||||
{
|
||||
systemctl stop etcd.service
|
||||
}
|
|
@ -0,0 +1,117 @@
|
|||
#!/usr/bin/env bash
|
||||
# Licensed to the Apache Software Foundation (ASF) under one
|
||||
# or more contributor license agreements. See the NOTICE file
|
||||
# distributed with this work for additional information
|
||||
# regarding copyright ownership. The ASF licenses this file
|
||||
# to you under the Apache License, Version 2.0 (the
|
||||
# "License"); you may not use this file except in compliance
|
||||
# with the License. You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
## @description install yarn
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function install_yarn()
|
||||
{
|
||||
install_yarn_container_executor
|
||||
install_yarn_config
|
||||
}
|
||||
|
||||
## @description uninstall yarn
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function uninstall_yarn()
|
||||
{
|
||||
rm -rf /etc/yarn/sbin/Linux-amd64-64/*
|
||||
rm -rf /etc/yarn/sbin/etc/hadoop/*
|
||||
}
|
||||
|
||||
## @description download yarn container executor
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function download_yarn_container_executor()
|
||||
{
|
||||
# my download http server
|
||||
if [[ -n "$DOWNLOAD_HTTP" ]]; then
|
||||
MY_YARN_CONTAINER_EXECUTOR_PATH="${DOWNLOAD_HTTP}/downloads/hadoop/container-executor"
|
||||
else
|
||||
MY_YARN_CONTAINER_EXECUTOR_PATH=${YARN_CONTAINER_EXECUTOR_PATH}
|
||||
fi
|
||||
|
||||
if [ ! -d "${DOWNLOAD_DIR}/hadoop" ]; then
|
||||
mkdir -p ${DOWNLOAD_DIR}/hadoop
|
||||
fi
|
||||
|
||||
if [[ -f "${DOWNLOAD_DIR}/hadoop/container-executor" ]]; then
|
||||
echo "${DOWNLOAD_DIR}/hadoop/container-executor is exist."
|
||||
else
|
||||
if [[ -n "$DOWNLOAD_HTTP" ]]; then
|
||||
echo "download ${MY_YARN_CONTAINER_EXECUTOR_PATH} ..."
|
||||
wget -P ${DOWNLOAD_DIR}/hadoop ${MY_YARN_CONTAINER_EXECUTOR_PATH}
|
||||
else
|
||||
echo "copy ${MY_YARN_CONTAINER_EXECUTOR_PATH} ..."
|
||||
cp ${MY_YARN_CONTAINER_EXECUTOR_PATH} ${DOWNLOAD_DIR}/hadoop/
|
||||
fi
|
||||
fi
|
||||
}
|
||||
|
||||
## @description install yarn container executor
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function install_yarn_container_executor()
|
||||
{
|
||||
echo "install yarn container executor file ..."
|
||||
|
||||
download_yarn_container_executor
|
||||
|
||||
if [ ! -d "/etc/yarn/sbin/Linux-amd64-64" ]; then
|
||||
mkdir -p /etc/yarn/sbin/Linux-amd64-64
|
||||
fi
|
||||
if [ -f "/etc/yarn/sbin/Linux-amd64-64/container-executor" ]; then
|
||||
rm /etc/yarn/sbin/Linux-amd64-64/container-executor
|
||||
fi
|
||||
|
||||
cp -f ${DOWNLOAD_DIR}/hadoop/container-executor /etc/yarn/sbin/Linux-amd64-64
|
||||
|
||||
sudo chmod 6755 /etc/yarn/sbin/Linux-amd64-64
|
||||
sudo chown :yarn /etc/yarn/sbin/Linux-amd64-64/container-executor
|
||||
sudo chmod 6050 /etc/yarn/sbin/Linux-amd64-64/container-executor
|
||||
}
|
||||
|
||||
## @description install yarn config
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function install_yarn_config()
|
||||
{
|
||||
echo "install yarn config file ..."
|
||||
|
||||
cp -R ${PACKAGE_DIR}/hadoop ${INSTALL_TEMP_DIR}/
|
||||
|
||||
find="/"
|
||||
replace="\/"
|
||||
escape_yarn_nodemanager_local_dirs=${YARN_NODEMANAGER_LOCAL_DIRS//$find/$replace}
|
||||
escape_yarn_nodemanager_log_dirs=${YARN_NODEMANAGER_LOG_DIRS//$find/$replace}
|
||||
escape_yarn_hierarchy=${YARN_HIERARCHY//$find/$replace}
|
||||
|
||||
sed -i "s/YARN_NODEMANAGER_LOCAL_DIRS_REPLACE/${escape_yarn_nodemanager_local_dirs}/g" $INSTALL_TEMP_DIR/hadoop/container-executor.cfg >>$LOG
|
||||
sed -i "s/YARN_NODEMANAGER_LOG_DIRS_REPLACE/${escape_yarn_nodemanager_log_dirs}/g" $INSTALL_TEMP_DIR/hadoop/container-executor.cfg >>$LOG
|
||||
sed -i "s/DOCKER_REGISTRY_REPLACE/${DOCKER_REGISTRY}/g" $INSTALL_TEMP_DIR/hadoop/container-executor.cfg >>$LOG
|
||||
sed -i "s/CALICO_NETWORK_NAME_REPLACE/${CALICO_NETWORK_NAME}/g" $INSTALL_TEMP_DIR/hadoop/container-executor.cfg >>$LOG
|
||||
sed -i "s/YARN_HIERARCHY_REPLACE/${escape_yarn_hierarchy}/g" $INSTALL_TEMP_DIR/hadoop/container-executor.cfg >>$LOG
|
||||
|
||||
# Delete the ASF license comment in the container-executor.cfg file, otherwise it will cause a cfg format error.
|
||||
sed -i '1,16d' $INSTALL_TEMP_DIR/hadoop/container-executor.cfg
|
||||
|
||||
if [ ! -d "/etc/yarn/sbin/etc/hadoop" ]; then
|
||||
mkdir -p /etc/yarn/sbin/etc/hadoop
|
||||
fi
|
||||
|
||||
cp -f $INSTALL_TEMP_DIR/hadoop/container-executor.cfg /etc/yarn/sbin/etc/hadoop/
|
||||
}
|
|
@ -0,0 +1,444 @@
|
|||
#!/usr/bin/env bash
|
||||
# Licensed to the Apache Software Foundation (ASF) under one
|
||||
# or more contributor license agreements. See the NOTICE file
|
||||
# distributed with this work for additional information
|
||||
# regarding copyright ownership. The ASF licenses this file
|
||||
# to you under the Apache License, Version 2.0 (the
|
||||
# "License"); you may not use this file except in compliance
|
||||
# with the License. You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
## @description main menu
|
||||
## @audience public
|
||||
## @stability stable
|
||||
main_menu()
|
||||
{
|
||||
cat<<MENULIST
|
||||
====================================================================================
|
||||
SUBMARINE INSTALLER ${SUBMARINE_INSTALLER_VERSION}
|
||||
|
||||
HOST:${LOCAL_HOST_IP} DOWNLOAD_SERVER:http://${DOWNLOAD_SERVER_IP}:${DOWNLOAD_SERVER_PORT}
|
||||
====================================================================================
|
||||
[Main menu]
|
||||
------------------------------------------------------------------------------------
|
||||
MENULIST
|
||||
echo -e " \e[32m1.prepare system environment [..]\e[0m"
|
||||
echo -e " \e[32m2.install component [..]\e[0m"
|
||||
echo -e " \e[32m3.uninstall component [..]\e[0m"
|
||||
echo -e " \e[32m4.start component [..]\e[0m"
|
||||
echo -e " \e[32m5.stop component [..]\e[0m"
|
||||
echo -e " \e[32m6.start download server [..]\e[0m"
|
||||
echo -e ""
|
||||
echo -e " \e[32mq.quit\e[0m"
|
||||
cat<<MENULIST
|
||||
====================================================================================
|
||||
MENULIST
|
||||
|
||||
echo -ne "Please input your choice [\e[32m1\e[0m-\e[32m6\e[0m,\e[32mq\e[0m(quit)]:"
|
||||
}
|
||||
|
||||
## @description check menu
|
||||
## @audience public
|
||||
## @stability stable
|
||||
check_menu()
|
||||
{
|
||||
cat<<MENULIST
|
||||
====================================================================================
|
||||
SUBMARINE INSTALLER ${SUBMARINE_INSTALLER_VERSION}
|
||||
|
||||
HOST:${LOCAL_HOST_IP} DOWNLOAD_SERVER:http://${DOWNLOAD_SERVER_IP}:${DOWNLOAD_SERVER_PORT}
|
||||
====================================================================================
|
||||
[Main menu] > [prepare system environment]
|
||||
------------------------------------------------------------------------------------
|
||||
MENULIST
|
||||
echo -e " \e[32m1.prepare operation system\e[0m"
|
||||
echo -e " \e[32m2.prepare operation system kernel\e[0m"
|
||||
echo -e " \e[32m3.prepare GCC version\e[0m"
|
||||
echo -e " \e[32m4.check GPU\e[0m"
|
||||
echo -e " \e[32m5.prepare user&group\e[0m"
|
||||
echo -e " \e[32m6.prepare nvidia environment\e[0m"
|
||||
echo -e ""
|
||||
echo -e " \e[32mb.back main menu\e[0m"
|
||||
cat<<MENULIST
|
||||
====================================================================================
|
||||
MENULIST
|
||||
|
||||
echo -ne "Please input your choice [\e[32m1\e[0m-\e[32m5\e[0m,\e[32mb\e[0m(back)]:"
|
||||
}
|
||||
|
||||
## @description install menu
|
||||
## @audience public
|
||||
## @stability stable
|
||||
install_menu()
|
||||
{
|
||||
cat<<MENULIST
|
||||
====================================================================================
|
||||
SUBMARINE INSTALLER ${SUBMARINE_INSTALLER_VERSION}
|
||||
|
||||
HOST:${LOCAL_HOST_IP} DOWNLOAD_SERVER:http://${DOWNLOAD_SERVER_IP}:${DOWNLOAD_SERVER_PORT}
|
||||
====================================================================================
|
||||
[Main menu] > [install component]
|
||||
------------------------------------------------------------------------------------
|
||||
MENULIST
|
||||
echo -e " \e[32m1.instll etcd\e[0m"
|
||||
echo -e " \e[32m2.instll docker\e[0m"
|
||||
echo -e " \e[32m3.instll calico network\e[0m"
|
||||
echo -e " \e[32m4.instll nvidia driver\e[0m"
|
||||
echo -e " \e[32m5.instll nvidia docker\e[0m"
|
||||
echo -e " \e[32m6.instll yarn container-executor\e[0m"
|
||||
echo -e " \e[32m7.instll submarine autorun script\e[0m"
|
||||
echo -e ""
|
||||
echo -e " \e[32mb.back main menu\e[0m"
|
||||
cat<<MENULIST
|
||||
====================================================================================
|
||||
MENULIST
|
||||
|
||||
echo -ne "Please input your choice [\e[32m1\e[0m-\e[32m7\e[0m,\e[32mb\e[0m(back)]:"
|
||||
}
|
||||
|
||||
## @description unstall menu
|
||||
## @audience public
|
||||
## @stability stable
|
||||
uninstall_menu()
|
||||
{
|
||||
cat<<MENULIST
|
||||
====================================================================================
|
||||
SUBMARINE INSTALLER ${SUBMARINE_INSTALLER_VERSION}
|
||||
|
||||
HOST:${LOCAL_HOST_IP} DOWNLOAD_SERVER:http://${DOWNLOAD_SERVER_IP}:${DOWNLOAD_SERVER_PORT}
|
||||
====================================================================================
|
||||
[Main menu] > [uninstll component]
|
||||
------------------------------------------------------------------------------------
|
||||
MENULIST
|
||||
echo -e " \e[32m1.uninstll etcd\e[0m"
|
||||
echo -e " \e[32m2.uninstll docker\e[0m"
|
||||
echo -e " \e[32m3.uninstll calico network\e[0m"
|
||||
echo -e " \e[32m4.uninstll nvidia driver\e[0m"
|
||||
echo -e " \e[32m5.uninstll nvidia docker\e[0m"
|
||||
echo -e " \e[32m6.uninstll yarn container-executor\e[0m"
|
||||
echo -e " \e[32m7.uninstll submarine autorun script\e[0m"
|
||||
echo -e ""
|
||||
echo -e " \e[32mb.back main menu\e[0m"
|
||||
cat<<MENULIST
|
||||
====================================================================================
|
||||
MENULIST
|
||||
|
||||
echo -ne "Please input your choice [\e[32m1\e[0m-\e[32m7\e[0m,\e[32mb\e[0m(back)]:"
|
||||
}
|
||||
|
||||
## @description start menu
|
||||
## @audience public
|
||||
## @stability stable
|
||||
start_menu()
|
||||
{
|
||||
cat<<MENULIST
|
||||
====================================================================================
|
||||
SUBMARINE INSTALLER ${SUBMARINE_INSTALLER_VERSION}
|
||||
|
||||
HOST:${LOCAL_HOST_IP} DOWNLOAD_SERVER:http://${DOWNLOAD_SERVER_IP}:${DOWNLOAD_SERVER_PORT}
|
||||
====================================================================================
|
||||
[Main menu] > [stop component]
|
||||
------------------------------------------------------------------------------------
|
||||
MENULIST
|
||||
echo -e " \e[32m1.start etcd\e[0m"
|
||||
echo -e " \e[32m2.start docker\e[0m"
|
||||
echo -e " \e[32m3.start calico network\e[0m"
|
||||
echo -e ""
|
||||
echo -e " \e[32mb.back main menu\e[0m"
|
||||
cat<<MENULIST
|
||||
====================================================================================
|
||||
MENULIST
|
||||
|
||||
echo -ne "Please input your choice [\e[32m1\e[0m-\e[32m3\e[0m,\e[32mb\e[0m(back)]:"
|
||||
}
|
||||
|
||||
## @description stop menu
|
||||
## @audience public
|
||||
## @stability stable
|
||||
stop_menu()
|
||||
{
|
||||
cat<<MENULIST
|
||||
====================================================================================
|
||||
SUBMARINE INSTALLER ${SUBMARINE_INSTALLER_VERSION}
|
||||
|
||||
HOST:${LOCAL_HOST_IP} DOWNLOAD_SERVER:http://${DOWNLOAD_SERVER_IP}:${DOWNLOAD_SERVER_PORT}
|
||||
====================================================================================
|
||||
[Main menu] > [stop component]
|
||||
------------------------------------------------------------------------------------
|
||||
MENULIST
|
||||
echo -e " \e[32m1.stop etcd\e[0m"
|
||||
echo -e " \e[32m2.stop docker\e[0m"
|
||||
echo -e " \e[32m3.stop calico network\e[0m"
|
||||
echo -e ""
|
||||
echo -e " \e[32mb.back main menu\e[0m"
|
||||
cat<<MENULIST
|
||||
====================================================================================
|
||||
MENULIST
|
||||
|
||||
echo -ne "Please input your choice [\e[32m1\e[0m-\e[32m3\e[0m,\e[32mb\e[0m(back)]:"
|
||||
}
|
||||
|
||||
## @description menu operation
|
||||
## @audience public
|
||||
## @stability stable
|
||||
menu_index="0"
|
||||
menu()
|
||||
{
|
||||
clear
|
||||
# echo "menu_index-menu_choice=$menu_index-$menu_choice"
|
||||
case $menu_index in
|
||||
"0")
|
||||
main_menu
|
||||
;;
|
||||
"1")
|
||||
check_menu
|
||||
;;
|
||||
"2")
|
||||
install_menu
|
||||
;;
|
||||
"3")
|
||||
uninstall_menu
|
||||
;;
|
||||
"4")
|
||||
start_menu
|
||||
;;
|
||||
"5")
|
||||
stop_menu
|
||||
;;
|
||||
"6")
|
||||
start_download_server
|
||||
;;
|
||||
"q")
|
||||
exit 1
|
||||
;;
|
||||
*)
|
||||
echo "error input!"
|
||||
menu_index="0"
|
||||
menu_choice="0"
|
||||
main_menu
|
||||
;;
|
||||
esac
|
||||
|
||||
read menu_choice
|
||||
}
|
||||
|
||||
## @description menu process
|
||||
## @audience public
|
||||
## @stability stable
|
||||
menu_process()
|
||||
{
|
||||
process=0
|
||||
unset myselect
|
||||
# echo "debug=$menu_index-$menu_choice"
|
||||
case "$menu_index-$menu_choice" in
|
||||
"1-b"|"2-b"|"3-b"|"4-b"|"5-b"|"6-b")
|
||||
menu_index="0"
|
||||
menu_choice="0"
|
||||
;;
|
||||
# check system environment
|
||||
"1-1")
|
||||
myselect="y"
|
||||
check_operationSystem
|
||||
;;
|
||||
"1-2")
|
||||
myselect="y"
|
||||
check_operationSystemKernel
|
||||
;;
|
||||
"1-3")
|
||||
myselect="y"
|
||||
check_gccVersion
|
||||
;;
|
||||
"1-4")
|
||||
myselect="y"
|
||||
check_GPU
|
||||
;;
|
||||
"1-5")
|
||||
myselect="y"
|
||||
check_userGroup
|
||||
;;
|
||||
"1-6")
|
||||
myselect="y"
|
||||
prepare_nvidia_environment
|
||||
;;
|
||||
# install component
|
||||
"2-1")
|
||||
echo -n "Do you want to install etcd?[y|n]"
|
||||
read myselect
|
||||
if [[ "$myselect" = "y" || "$myselect" = "Y" ]]
|
||||
then
|
||||
install_etcd
|
||||
fi
|
||||
;;
|
||||
"2-2")
|
||||
echo -n "Do you want to install docker?[y|n]"
|
||||
read myselect
|
||||
if [[ "$myselect" = "y" || "$myselect" = "Y" ]]
|
||||
then
|
||||
install_docker
|
||||
fi
|
||||
;;
|
||||
"2-3")
|
||||
echo -n "Do you want to install calico network?[y|n]"
|
||||
read myselect
|
||||
if [[ "$myselect" = "y" || "$myselect" = "Y" ]]
|
||||
then
|
||||
install_calico
|
||||
fi
|
||||
;;
|
||||
"2-4")
|
||||
echo -n "Do you want to install nvidia driver?[y|n]"
|
||||
read myselect
|
||||
if [[ "$myselect" = "y" || "$myselect" = "Y" ]]
|
||||
then
|
||||
install_nvidia
|
||||
fi
|
||||
;;
|
||||
"2-5")
|
||||
echo -n "Do you want to install nvidia docker?[y|n]"
|
||||
read myselect
|
||||
if [[ "$myselect" = "y" || "$myselect" = "Y" ]]
|
||||
then
|
||||
install_nvidia_docker
|
||||
fi
|
||||
;;
|
||||
"2-6")
|
||||
echo -n "Do you want to install yarn container-executor?[y|n]"
|
||||
read myselect
|
||||
if [[ "$myselect" = "y" || "$myselect" = "Y" ]]
|
||||
then
|
||||
install_yarn
|
||||
fi
|
||||
;;
|
||||
"2-7")
|
||||
echo -n "Do you want to install submarine auto start script?[y|n]"
|
||||
read myselect
|
||||
if [[ "$myselect" = "y" || "$myselect" = "Y" ]]
|
||||
then
|
||||
install_submarine
|
||||
fi
|
||||
;;
|
||||
# uninstall component
|
||||
"3-1")
|
||||
echo -n "Do you want to uninstall etcd?[y|n]"
|
||||
read myselect
|
||||
if [[ "$myselect" = "y" || "$myselect" = "Y" ]]
|
||||
then
|
||||
uninstall_etcd
|
||||
fi
|
||||
;;
|
||||
"3-2")
|
||||
echo -n "Do you want to uninstall docker?[y|n]"
|
||||
read myselect
|
||||
if [[ "$myselect" = "y" || "$myselect" = "Y" ]]
|
||||
then
|
||||
uninstall_docker
|
||||
fi
|
||||
;;
|
||||
"3-3")
|
||||
echo -n "Do you want to uninstall calico network?[y|n]"
|
||||
read myselect
|
||||
if [[ "$myselect" = "y" || "$myselect" = "Y" ]]
|
||||
then
|
||||
uninstall_calico
|
||||
fi
|
||||
;;
|
||||
"3-4")
|
||||
echo -n "Do you want to uninstall nvidia driver?[y|n]"
|
||||
read myselect
|
||||
if [[ "$myselect" = "y" || "$myselect" = "Y" ]]
|
||||
then
|
||||
uninstall_nvidia
|
||||
fi
|
||||
;;
|
||||
"3-5")
|
||||
echo -n "Do you want to uninstall nvidia docker?[y|n]"
|
||||
read myselect
|
||||
if [[ "$myselect" = "y" || "$myselect" = "Y" ]]
|
||||
then
|
||||
uninstall_nvidia_docker
|
||||
fi
|
||||
;;
|
||||
"3-6")
|
||||
echo -n "Do you want to uninstall yarn container-executor?[y|n]"
|
||||
read myselect
|
||||
if [[ "$myselect" = "y" || "$myselect" = "Y" ]]
|
||||
then
|
||||
uninstall_yarn
|
||||
fi
|
||||
;;
|
||||
"3-7")
|
||||
echo -n "Do you want to uninstall submarine autostart script?[y|n]"
|
||||
read myselect
|
||||
if [[ "$myselect" = "y" || "$myselect" = "Y" ]]
|
||||
then
|
||||
uninstall_submarine
|
||||
fi
|
||||
;;
|
||||
# startup component
|
||||
"4-1")
|
||||
echo -n "Do you want to startup etcd?[y|n]"
|
||||
read myselect
|
||||
if [[ "$myselect" = "y" || "$myselect" = "Y" ]]
|
||||
then
|
||||
start_etcd
|
||||
fi
|
||||
;;
|
||||
"4-2")
|
||||
echo -n "Do you want to startup docker?[y|n]"
|
||||
read myselect
|
||||
if [[ "$myselect" = "y" || "$myselect" = "Y" ]]
|
||||
then
|
||||
start_docker
|
||||
fi
|
||||
;;
|
||||
"4-3")
|
||||
echo -n "Do you want to startup calico network?[y|n]"
|
||||
read myselect
|
||||
if [[ "$myselect" = "y" || "$myselect" = "Y" ]]
|
||||
then
|
||||
start_calico
|
||||
fi
|
||||
;;
|
||||
# stop component
|
||||
"5-1")
|
||||
echo -n "Do you want to stop etcd?[y|n]"
|
||||
read myselect
|
||||
if [[ "$myselect" = "y" || "$myselect" = "Y" ]]
|
||||
then
|
||||
stop_etcd
|
||||
fi
|
||||
;;
|
||||
"5-2")
|
||||
echo -n "Do you want to stop docker?[y|n]"
|
||||
read myselect
|
||||
if [[ "$myselect" = "y" || "$myselect" = "Y" ]]
|
||||
then
|
||||
stop_docker
|
||||
fi
|
||||
;;
|
||||
"5-3")
|
||||
echo -n "Do you want to stop calico network?[y|n]"
|
||||
read myselect
|
||||
if [[ "$myselect" = "y" || "$myselect" = "Y" ]]
|
||||
then
|
||||
stop_calico
|
||||
fi
|
||||
;;
|
||||
esac
|
||||
|
||||
if [[ "$myselect" = "y" || "$myselect" = "Y" ]]
|
||||
then
|
||||
process=1
|
||||
fi
|
||||
|
||||
# echo "process=$process"
|
||||
return $process
|
||||
}
|
||||
|
|
@ -0,0 +1,99 @@
|
|||
#!/usr/bin/env bash
|
||||
# Licensed to the Apache Software Foundation (ASF) under one
|
||||
# or more contributor license agreements. See the NOTICE file
|
||||
# distributed with this work for additional information
|
||||
# regarding copyright ownership. The ASF licenses this file
|
||||
# to you under the Apache License, Version 2.0 (the
|
||||
# "License"); you may not use this file except in compliance
|
||||
# with the License. You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
## @description download nvidia docker bin
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function download_nvidia_docker_bin()
|
||||
{
|
||||
# download http server
|
||||
if [[ -n "$DOWNLOAD_HTTP" ]]; then
|
||||
MY_NVIDIA_DOCKER_RPM_URL="${DOWNLOAD_HTTP}/downloads/nvidia-docker/${NVIDIA_DOCKER_RPM}"
|
||||
else
|
||||
MY_NVIDIA_DOCKER_RPM_URL=${NVIDIA_DOCKER_RPM_URL}
|
||||
fi
|
||||
|
||||
if [[ -f "${DOWNLOAD_DIR}/nvidia-docker/${NVIDIA_DOCKER_RPM}" ]]; then
|
||||
echo "${DOWNLOAD_DIR}/nvidia-docker/${NVIDIA_DOCKER_RPM} is exist."
|
||||
else
|
||||
echo "download ${MY_NVIDIA_DOCKER_RPM_URL} ..."
|
||||
wget -P ${DOWNLOAD_DIR}/nvidia-docker/ ${MY_NVIDIA_DOCKER_RPM_URL}
|
||||
fi
|
||||
}
|
||||
|
||||
## @description install nvidia docker
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function install_nvidia_docker()
|
||||
{
|
||||
download_nvidia_docker_bin
|
||||
|
||||
sudo rpm -i ${DOWNLOAD_DIR}/nvidia-docker/${NVIDIA_DOCKER_RPM}
|
||||
|
||||
echo -e "\033[32m===== Start nvidia-docker =====\033[0m"
|
||||
sudo systemctl start nvidia-docker
|
||||
|
||||
echo -e "\033[32m===== Check nvidia-docker status =====\033[0m"
|
||||
systemctl status nvidia-docker
|
||||
|
||||
echo -e "\033[32m===== Check nvidia-docker log =====\033[0m"
|
||||
journalctl -u nvidia-docker
|
||||
|
||||
echo -e "\033[32m===== Test nvidia-docker-plugin =====\033[0m"
|
||||
curl http://localhost:3476/v1.0/docker/cli
|
||||
|
||||
# create nvidia driver library path
|
||||
if [ ! -d "/var/lib/nvidia-docker/volumes/nvidia_driver" ]; then
|
||||
echo "WARN: /var/lib/nvidia-docker/volumes/nvidia_driver folder path is not exist!"
|
||||
mkdir -p /var/lib/nvidia-docker/volumes/nvidia_driver
|
||||
fi
|
||||
|
||||
local nvidiaVersion=`get_nvidia_version`
|
||||
echo -e "\033[31m nvidia detect version is ${nvidiaVersion}\033[0m"
|
||||
|
||||
mkdir /var/lib/nvidia-docker/volumes/nvidia_driver/${nvidiaVersion}
|
||||
mkdir /var/lib/nvidia-docker/volumes/nvidia_driver/${nvidiaVersion}/bin
|
||||
mkdir /var/lib/nvidia-docker/volumes/nvidia_driver/${nvidiaVersion}/lib64
|
||||
|
||||
cp /usr/bin/nvidia* /var/lib/nvidia-docker/volumes/nvidia_driver/${nvidiaVersion}/bin
|
||||
cp /usr/lib64/libcuda* /var/lib/nvidia-docker/volumes/nvidia_driver/${nvidiaVersion}/lib64
|
||||
cp /usr/lib64/libnvidia* /var/lib/nvidia-docker/volumes/nvidia_driver/${nvidiaVersion}/lib64
|
||||
|
||||
echo -e "\033[32m===== Please manually execute the following command =====\033[0m"
|
||||
echo -e "\033[32mshell:> nvidia-docker run --rm ${DOCKER_REGISTRY}/nvidia/cuda:9.0-devel nvidia-smi
|
||||
# If you don't see the list of graphics cards above, the NVIDIA driver installation failed. =====
|
||||
\033[0m"
|
||||
|
||||
echo -e "\033[32m===== Please manually execute the following command =====\033[0m"
|
||||
echo -e "\033[32m# Test with tf.test.is_gpu_available()
|
||||
shell:> nvidia-docker run -it ${DOCKER_REGISTRY}/tensorflow/tensorflow:1.9.0-gpu bash
|
||||
# In docker container
|
||||
container:> python
|
||||
python:> import tensorflow as tf
|
||||
python:> tf.test.is_gpu_available()
|
||||
python:> exit()
|
||||
\033[0m"
|
||||
}
|
||||
|
||||
## @description uninstall nvidia docker
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function uninstall_nvidia_docker()
|
||||
{
|
||||
echo "This method is not implemented."
|
||||
}
|
||||
|
|
@ -0,0 +1,120 @@
|
|||
#!/usr/bin/env bash
|
||||
# Licensed to the Apache Software Foundation (ASF) under one
|
||||
# or more contributor license agreements. See the NOTICE file
|
||||
# distributed with this work for additional information
|
||||
# regarding copyright ownership. The ASF licenses this file
|
||||
# to you under the Apache License, Version 2.0 (the
|
||||
# "License"); you may not use this file except in compliance
|
||||
# with the License. You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
nvidia_run_file=""
|
||||
|
||||
## @description get nvidia version
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function get_nvidia_version()
|
||||
{
|
||||
chmod +x ${DOWNLOAD_DIR}/nvidia/nvidia-detect
|
||||
local nvidia_detect_info=`${DOWNLOAD_DIR}/nvidia/nvidia-detect -v`
|
||||
echo $nvidia_detect_info | sed "s/^.*This device requires the current \([0-9.]*\).*/\1/"
|
||||
}
|
||||
|
||||
## @description download nvidia driver
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function download_nvidia_driver()
|
||||
{
|
||||
# submarin http server
|
||||
if [[ -n "$DOWNLOAD_HTTP" ]]; then
|
||||
MY_NVIDIA_DETECT_URL="${DOWNLOAD_HTTP}/downloads/nvidia/nvidia-detect"
|
||||
else
|
||||
MY_NVIDIA_DETECT_URL=${NVIDIA_DETECT_URL}
|
||||
fi
|
||||
|
||||
# download nvidia_detect
|
||||
if [[ -f "${DOWNLOAD_DIR}/nvidia/nvidia-detect" ]]; then
|
||||
echo "${DOWNLOAD_DIR}/nvidia/nvidia-detect is exist."
|
||||
else
|
||||
echo "download ${MY_NVIDIA_DETECT_URL} ..."
|
||||
wget -P ${DOWNLOAD_DIR}/nvidia/ ${MY_NVIDIA_DETECT_URL}
|
||||
fi
|
||||
|
||||
echo "execution nvidia-detect to check the graphics card ..."
|
||||
local nvidiaVersion=`get_nvidia_version`
|
||||
echo -e "detect nvidia version is \033[31m${nvidiaVersion}\033[0m"
|
||||
|
||||
# download NVIDIA driver
|
||||
if [[ "$nvidiaVersion" = "" ]]; then
|
||||
echo -e "\033[31mERROR: No graphics card device detected.\033[0m"
|
||||
return 1
|
||||
else
|
||||
nvidia_run_file="NVIDIA-Linux-x86_64-${nvidiaVersion}.run"
|
||||
|
||||
# submarin http server
|
||||
if [[ -n "$DOWNLOAD_HTTP" ]]; then
|
||||
MY_NVIDIA_DRIVER_RUN_URL="${DOWNLOAD_HTTP}/downloads/nvidia/${nvidia_run_file}"
|
||||
else
|
||||
# http://us.download.nvidia.com/XFree86/Linux-x86_64/390.87/NVIDIA-Linux-x86_64-390.87.run
|
||||
MY_NVIDIA_DRIVER_RUN_URL="http://us.download.nvidia.com/XFree86/Linux-x86_64/${nvidiaVersion}/${nvidia_run_file}"
|
||||
fi
|
||||
|
||||
if [[ -f ${DOWNLOAD_DIR}/nvidia/${nvidia_run_file} ]]; then
|
||||
echo "NVIDIA driver files already exist in the ${DOWNLOAD_DIR}/nvidia/${nvidia_run_file} directory."
|
||||
echo "===== Please make sure the ${DOWNLOAD_DIR}/nvidia/nvidia/${nvidia_run_file} file is complete and can be used normally. ====="
|
||||
else
|
||||
echo "Download the NVIDIA driver from the ${MY_NVIDIA_DRIVER_RUN_URL}"
|
||||
wget -P ${DOWNLOAD_DIR}/nvidia/ ${MY_NVIDIA_DRIVER_RUN_URL}
|
||||
fi
|
||||
fi
|
||||
}
|
||||
|
||||
## @description install nvidia
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function install_nvidia()
|
||||
{
|
||||
download_nvidia_driver
|
||||
|
||||
# Confirm that the system disables nouveau
|
||||
local disable_nouveau_info=`lsmod | grep nouveau`
|
||||
if [[ "$disable_nouveau_info" = "" ]]; then
|
||||
echo "===== Start installing the NVIDIA driver ====="
|
||||
echo -e "Some options during the installation
|
||||
Would you like to register the kernel module sources with DKMS?
|
||||
This will allow DKMS to automatically build a new module, if you install a different kernel later. \033[33m[Yes]\033[0m
|
||||
Install NVIDIA's 32-bit compatibility libraries \033[33m[Yes]\033[0m
|
||||
centos Install NVIDIA's 32-bit compatibility libraries \033[33m[Yes]\033[0m
|
||||
Would you like to run the nvidia-xconfig utility to automatically update your X configuration file... \033[33m[No]\033[0m"
|
||||
sleep 2
|
||||
sh ${DOWNLOAD_DIR}/nvidia/${nvidia_run_file}
|
||||
else
|
||||
echo -e "ERROR: Nouveau is not disabled"
|
||||
return 1
|
||||
fi
|
||||
|
||||
echo -e "\033[32m===== execute nvidia-smi. You should be able to see the list of graphics cards =====\033[0m"
|
||||
sleep 1
|
||||
nvidia-smi
|
||||
}
|
||||
|
||||
## @description uninstall nvidia
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function uninstall_nvidia()
|
||||
{
|
||||
if [ ! -f "/usr/bin/nvidia-uninstall" ]; then
|
||||
echo -e "\033[31mERROR: /usr/bin/nvidia-uninstall file is not exist!\033[0m"
|
||||
return 1
|
||||
fi
|
||||
|
||||
echo -e "execute /usr/bin/nvidia-uninstall"
|
||||
/usr/bin/nvidia-uninstall
|
||||
}
|
|
@ -0,0 +1,38 @@
|
|||
#!/usr/bin/env bash
|
||||
# Licensed to the Apache Software Foundation (ASF) under one
|
||||
# or more contributor license agreements. See the NOTICE file
|
||||
# distributed with this work for additional information
|
||||
# regarding copyright ownership. The ASF licenses this file
|
||||
# to you under the Apache License, Version 2.0 (the
|
||||
# "License"); you may not use this file except in compliance
|
||||
# with the License. You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
## @description install submarine autorun script
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function install_submarine()
|
||||
{
|
||||
cp ${PACKAGE_DIR}/submarine/submarine.sh /etc/rc.d/init.d/submarine.sh
|
||||
chmod +x /etc/rc.d/init.d/submarine.sh
|
||||
chmod +x /etc/rc.d/rc.local
|
||||
|
||||
if [ `grep -c "/etc/rc.d/init.d/submarine.sh" /etc/rc.d/rc.local` -eq '0' ]; then
|
||||
echo "/etc/rc.d/init.d/submarine.sh">> /etc/rc.d/rc.local
|
||||
fi
|
||||
}
|
||||
|
||||
## @description uninstall submarine autorun script
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function uninstall_submarine()
|
||||
{
|
||||
rm /etc/rc.d/init.d/submarine.sh
|
||||
}
|
|
@ -0,0 +1,123 @@
|
|||
#!/usr/bin/env bash
|
||||
# Licensed to the Apache Software Foundation (ASF) under one
|
||||
# or more contributor license agreements. See the NOTICE file
|
||||
# distributed with this work for additional information
|
||||
# regarding copyright ownership. The ASF licenses this file
|
||||
# to you under the Apache License, Version 2.0 (the
|
||||
# "License"); you may not use this file except in compliance
|
||||
# with the License. You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
## @description check install user
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function check_install_user()
|
||||
{
|
||||
if [[ $(id -u) -ne 0 ]];then
|
||||
echo "This script must be run with a ROOT user!"
|
||||
exit # don't call exit_install()
|
||||
fi
|
||||
}
|
||||
|
||||
## @description exit install
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function exit_install()
|
||||
{
|
||||
echo "Exit the installation!" | tee -a $LOG
|
||||
exit $1
|
||||
}
|
||||
|
||||
## @description Check if the IP address format is correct
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function valid_ip()
|
||||
{
|
||||
local ip=$1
|
||||
local stat=1
|
||||
|
||||
if [[ $ip =~ ^[0-9]{1,3\}.[0-9]{1,3\}.[0-9]{1,3\}.[0-9]{1,3\}$ ]]; then
|
||||
OIFS=$IFS
|
||||
IFS='.'
|
||||
ip=($ip)
|
||||
IFS=$OIFS
|
||||
|
||||
if [[ ${ip[0]} -le 255 && ${ip[1]} -le 255 && ${ip[2]} -le 255 && ${ip[3]} -le 255 ]]; then
|
||||
stat=$?
|
||||
fi
|
||||
fi
|
||||
|
||||
return $stat
|
||||
}
|
||||
|
||||
## @description Check if the configuration file configuration is correct
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function check_install_conf()
|
||||
{
|
||||
echo "Check if the configuration file configuration is correct ..." | tee -a $LOG
|
||||
|
||||
# check etcd conf
|
||||
hostCount=${#ETCD_HOSTS[@]}
|
||||
if [[ $hostCount -lt 3 && hostCount -ne 0 ]]; then # <>2
|
||||
echo "Number of nodes = [$hostCount], must be configured to be greater than or equal to 3 servers! " | tee -a $LOG
|
||||
exit_install
|
||||
fi
|
||||
for ip in ${ETCD_HOSTS[@]}
|
||||
do
|
||||
if ! valid_ip $ip; then
|
||||
echo "]ETCD_HOSTS=[$ip], IP address format is incorrect! " | tee -a $LOG
|
||||
exit_install
|
||||
fi
|
||||
done
|
||||
echo "Check if the configuration file configuration is correct [ Done ]" | tee -a $LOG
|
||||
}
|
||||
|
||||
## @description index by EtcdHosts list
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function indexByEtcdHosts() {
|
||||
index=0
|
||||
while [ "$index" -lt "${#ETCD_HOSTS[@]}" ]; do
|
||||
if [ "${ETCD_HOSTS[$index]}" = "$1" ]; then
|
||||
echo $index
|
||||
return
|
||||
fi
|
||||
let "index++"
|
||||
done
|
||||
echo ""
|
||||
}
|
||||
|
||||
## @description get local IP
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function getLocalIP()
|
||||
{
|
||||
local _ip _myip _line _nl=$'\n'
|
||||
while IFS=$': \t' read -a _line ;do
|
||||
[ -z "${_line%inet}" ] &&
|
||||
_ip=${_line[${#_line[1]}>4?1:2]} &&
|
||||
[ "${_ip#127.0.0.1}" ] && _myip=$_ip
|
||||
done< <(LANG=C /sbin/ifconfig)
|
||||
printf ${1+-v} $1 "%s${_nl:0:$[${#1}>0?0:1]}" $_myip
|
||||
}
|
||||
|
||||
## @description get ip list
|
||||
## @audience public
|
||||
## @stability stable
|
||||
function get_ip_list()
|
||||
{
|
||||
array=$(ifconfig | grep inet | grep -v inet6 | grep -v 127 | sed 's/^[ \t]*//g' | cut -d ' ' -f2)
|
||||
|
||||
for ip in ${array[@]}
|
||||
do
|
||||
LOCAL_HOST_IP_LIST+=(${ip})
|
||||
done
|
||||
}
|
Loading…
Reference in New Issue