Run under virtualenv in AWS dynamic cluster management service

This commit is contained in:
allegroai 2020-01-14 11:44:20 +02:00
parent dee2475698
commit b1ca90a303

View File

@ -269,15 +269,18 @@
" # user_data script will automatically run when the instance is started. \n", " # user_data script will automatically run when the instance is started. \n",
" # It will install the required packages for trains-agent configure it using \n", " # It will install the required packages for trains-agent configure it using \n",
" # environment variables and run trains-agent on the required queue\n", " # environment variables and run trains-agent on the required queue\n",
" user_data = \"\"\"#!/bin/bash \n", " user_data = \"\"\"#!/bin/bash\n",
" sudo apt-get update\n", " sudo apt-get update\n",
" sudo apt-get install -y python3-dev\n", " sudo apt-get install -y python3-dev\n",
" sudo apt-get install -y python3-pip\n", " sudo apt-get install -y python3-pip\n",
" sudo apt-get install -y gcc\n", " sudo apt-get install -y gcc\n",
" sudo apt-get install -y git\n", " sudo apt-get install -y git\n",
" sudo apt-get install -y build-essential\n", " sudo apt-get install -y build-essential\n",
python3 -m pip install screen\n", " python3 -m pip install -U pip\n",
" sudo python3 -m pip install trains-agent\n", " python3 -m pip install virtualenv\n",
" python3 -m virtualenv trains_agent_venv\n",
" source trains_agent_venv/bin/activate\n",
" python -m pip install trains-agent\n",
" echo 'agent.git_user=\\\"{git_user}\\\"' >> /root/trains.conf\n", " echo 'agent.git_user=\\\"{git_user}\\\"' >> /root/trains.conf\n",
" echo 'agent.git_pass=\\\"{git_pass}\\\"' >> /root/trains.conf\n", " echo 'agent.git_pass=\\\"{git_pass}\\\"' >> /root/trains.conf\n",
" echo \"{trains_conf}\" >> /root/trains.conf\n", " echo \"{trains_conf}\" >> /root/trains.conf\n",
@ -288,9 +291,8 @@ python3 -m pip install screen\n",
" export TRAINS_WORKER_ID={worker_id}:$DYNAMIC_INSTANCE_ID\n", " export TRAINS_WORKER_ID={worker_id}:$DYNAMIC_INSTANCE_ID\n",
" export TRAINS_API_ACCESS_KEY='{access_key}'\n", " export TRAINS_API_ACCESS_KEY='{access_key}'\n",
" export TRAINS_API_SECRET_KEY='{secret_key}'\n", " export TRAINS_API_SECRET_KEY='{secret_key}'\n",
" screen\n",
" {bash_script}\n", " {bash_script}\n",
" python3 -m trains_agent --config-file '/root/trains.conf' daemon --queue '{queue}' {docker}\n", " python -m trains_agent --config-file '/root/trains.conf' daemon --queue '{queue}' {docker}\n",
" shutdown\n", " shutdown\n",
" \"\"\".format(\n", " \"\"\".format(\n",
" api_server=TRAINS_SERVER_API_SERVER,\n", " api_server=TRAINS_SERVER_API_SERVER,\n",