A library for helping to deploy a Numer.ai compute node
Project description
numerai-cli
This is a CLI for setting up a Numerai compute node and deploying your models to it. This sets up a compute cluster in AWS (Amazon Web Services), and is architected to cost a minimal amount of money to run (on average, you will spend less than $1 per month).
Questions or feedback? Join us at RocketChat
- Prerequisites
- Setup
- Quickstart
- Compute Node Architecture
- Docker Example Explained
- Commands
- Troubleshooting
- Prerequisites Help
- Uninstall
Prerequisites
All you need is:
- AWS (Amazon Web Services) account setup
- AWS API keys (Access key ID and Secret Access Key)
- Numerai API Key (Public ID and secret key)
Setup
For your convenience, we've included setup scripts in the scripts
directory that will ensure the prerequisites are installed.
You can download and run the setup script for your OS with one of the following commands:
-
Mac Terminal(cmd + space, type
terminal
, selectterminal.app
):curl https://raw.githubusercontent.com/numerai/numerai-cli/ndharasz/setup-scripts/scripts/setup-mac.sh | bash
-
Ubuntu 18 Terminal (ctrl + alt + t):
apt update && apt install -y libcurl4 curl && curl https://raw.githubusercontent.com/numerai/numerai-cli/ndharasz/setup-scripts/scripts/setup-ubu18.sh | bash
-
Windows Command Prompt (windows key, type
cmd
, select Command Prompt):powershell -command "$Script = Invoke-WebRequest 'https://raw.githubusercontent.com/numerai/numerai-cli/ndharasz/setup-scripts/scripts/setup-win10.ps1'; $ScriptBlock = [ScriptBlock]::Create($Script.Content); Invoke-Command -ScriptBlock $ScriptBlock"
See the Prerequisites Help section if you need further help getting these setup.
The CLI has been tested and found working on MacOS/OSX, Windows 10, and Ubuntu 18.04, but should theoretically work anywhere that docker and Python are available.
Install this library with:
pip3 install numerai-cli
Quickstart
The following instructions will get you setup with a compute node in a matter of minutes:
mkdir example-numerai
cd example-numerai
numerai setup
numerai docker copy-example
numerai docker deploy
numerai setup
will prompt your for AWS and Numerai API keys. Please refer to the AWS and Numerai API Key sections for instructions on how to obtain those.
Your compute node is now setup and ready to run. Look in the .numerai/submission_url.txt
file to see your submission url that you will provide to Numerai as your webhook url. Go to your Numerai account and select the "Compute" section to enter it there.
The default example does not stake, so you will still have to manually do that every week. Alternatively, check out the bottom of predict.py for example code on how to stake automatically.
Testing
You can test the webhook url directly like so:
numerai compute test-webhook
If the command succeeds, it will return quickly with a status of "pending". This means that your container has been scheduled to run but hasn't actually started yet.
You can check that your job has been scheduled by running:
numerai compute status
And once it's in the RUNNING
state, you can look at the logs:
numerai compute logs
You can also check for the raw AWS logs at https://console.aws.amazon.com/ecs/home?region=us-east-1#/clusters/numerai-submission-ecs-cluster/tasks or logs from your container at https://console.aws.amazon.com/cloudwatch/home?region=us-east-1#logStream:group=/fargate/service/numerai-submission
NOTE: the container takes a little time to schedule. The first time it runs also tends to take longer (2-3min), with subsequent runs starting a lot faster.
Local Testing
You can test your container locally by running:
numerai docker run
This will run the container exactly like it runs in the cloud. Keep in mind that this will be limited by your machine's RAM, or in the case of Windows/OSX, you can increase how much memory is allocated to docker: Windows or OSX
Common Problems
numerai
not in PATH
numerai: command not found
Try and run ~/.local/bin/numerai
on osx/linux or %LOCALAPPDATA%\Programs\Python\Python37-32\Scripts\numerai.exe
on Windows.
Alternatively, exit your terminal/command prompt and re-open it. By default, pip will try to add itself to your PATH for subsequent runs, but it requires you to restart the terminal.
Docker not installed
...
subprocess.CalledProcessError: Command 'docker run --rm -it -v /home/jason/tmp/.numerai:/opt/plan -w /opt/plan hashicorp/terraform:light init' returned non-zero exit status 127.
If you're certain that docker is installed, make sure that your user can execute docker, ie. try to run docker ps
. If that's the issue, then depending on your system, you can do the following:
Windows/OSX
Make sure the docker application is running and finished booting up. It can take a few minutes to be completely ready. When clicking on the docker tray icon, it should say "Docker Desktop is Running".
If you're using Docker Toolbox on Windows, then make sure you've opened the "Docker Quickstart Terminal".
Linux
sudo usermod -aG docker $USER
Then reboot or logout/login for this to take effect.
Wrong AWS API key
...
Error: error validating provider credentials: error calling sts:GetCallerIdentity: InvalidClientTokenId: The security token included in the request is invalid.
status code: 403, request id: 32500359-7d9e-11e9-b0ed-596aba1b72c5
...
subprocess.CalledProcessError: Command 'docker run -e "AWS_ACCESS_KEY_ID=..." -e "AWS_SECRET_ACCESS_KEY=..." --rm -it -v /home/jason/tmp/.numerai:/opt/plan -w /opt/plan hashicorp/terraform:light apply -auto-approve' returned non-zero exit status 1.
Drive not shared
If you get:
docker: Error response from daemon: Drive has not been shared
Then you need to share your drive. See https://docs.docker.com/docker-for-windows/#shared-drives for details.
Compute Node Architecture
numerai setup
under the hood uses Terraform to setup an AWS environment with the following:
- An ECR (Elastic Container Repository) for storing docker containers
- A Fargate task to run your compute job in the ECS (Elastic Container Service)
- A lambda endpoint that schedules your compute job to run
There's actually a bunch of other bits of glue in AWS that are setup to run this, but these 3 are the most important. The lambda endpoint corresponds to the submission url that your provide back to Numerai. The ECR is where numerai docker deploy
will push your image to. Fargate is where your task actually runs in the ECS, and it's where you'll want to look if things don't appear to be actually submitting.
Submission Webhook URL
This is the url that you provide back to Numerai. It's the thing that triggers the lambda and schedules your job to run. Once you've setup the webhook in your Numerai account, it will be called Saturday morning right after a new round opens, and if your job fails (determined by not having submitted all submissions successfully) then it will be triggered again around 24 hours later.
Docker example
Lets look at the docker example's files:
▶ tree
.
├── Dockerfile
├── model.py
├── predict.py
├── requirements.txt
└── train.py
Dockerfile
And then lets look at the Dockerfile:
FROM python:3
ADD requirements.txt .
RUN pip install -r requirements.txt
ADD . .
ARG NUMERAI_PUBLIC_ID
ENV NUMERAI_PUBLIC_ID=$NUMERAI_PUBLIC_ID
ARG NUMERAI_SECRET_KEY
ENV NUMERAI_SECRET_KEY=$NUMERAI_SECRET_KEY
CMD [ "python", "./predict.py" ]
So breaking this down line by line:
FROM python:3
This Dockerfile inherits from python:3
, which provides us a working Python environment.
ADD requirements.txt .
RUN pip install -r requirements.txt
We then add the requirements.txt file, and pip install every requirement from it. The ADD
keyword will take a file from your current directory and copy it over to the Docker container.
ADD . .
After that, we add everything in the current directory. This will include all of your code, as well as any other files such as serialized models that you've saved to the current directory.
ARG NUMERAI_PUBLIC_ID
ENV NUMERAI_PUBLIC_ID=$NUMERAI_PUBLIC_ID
ARG NUMERAI_SECRET_KEY
ENV NUMERAI_SECRET_KEY=$NUMERAI_SECRET_KEY
These are docker arguments that numerai train/run/deploy
will always pass into docker. They are then set in your environment, so that you can access them from your script like so:
import os
public_id = os.environ["NUMERAI_PUBLIC_ID"]
secret_key = os.environ["NUMERAI_SECRET_KEY"]
CMD [ "python", "./predict.py" ]
This sets the default command to run your docker container. This is overridden in the numerai docker train
command, but otherwise this will be the command that is always run when using numerai docker run
and numerai docker deploy
model.py
The model code lives in here. We're using numerox, but realistically this file could host any kind of model.
train.py
The code that gets run when running numerai docker train
predict.py
The code that gets run when running numerai docker run
and is deployed to run in the numerai compute node after executing numerai docker deploy
Commands
numerai setup
The following command will setup a full Numerai compute cluster in AWS:
numerai setup
If this is your first time running, it will also ask for your AWS and Numerai API keys. These keys are stored in $HOME/.numerai for future runs.
There will be a bunch of output about how it's setting up the AWS cluster, but the only important part is at the end:
...
Outputs:
docker_repo = 505651907052.dkr.ecr.us-east-1.amazonaws.com/numerai-submission
submission_url = https://wzq6vxvj8j.execute-api.us-east-1.amazonaws.com/v1/submit
- submission_url is your webhook url that you will provide to Numerai. Save this for later. If you forget it, a copy is stored in
.numerai/submission_url.txt
. - docker_repo will be used in the next step but you don't need to worry about it since it's all automated for you
This command is idempotent and safe to run multiple times.
numerai docker copy-example
If you don't have a model already setup, then you should copy over the docker example.
numerai docker copy-example
WARNING: this will overwrite the following files if they exist: Dockerfile, model.py, train.py, predict.py, and requirements.txt
If you manage multiple tournament models, you should copy the python3-multimodel
example
numerai docker copy-example --python3-multimodel
As well as using R:
numerai docker copy-example --rlang
numerai docker train (optional, but highly recommended)
Trains your model by running train.py
. This assumes a file called train.py
exists and serializes your model to this directory. See the example if you want inspiration for how to do this.
numerai docker train
numerai docker run (optional)
To test your docker container locally, you can run it with this command. This will run the default CMD for the Dockerfile, which for the default example is predict.py
.
numerai docker run
numerai docker deploy
Builds and pushes your docker image to the AWS docker repo
numerai docker deploy
numerai destroy
If you ever want to delete the AWS environment to save costs or start from scratch, you can run the following:
numerai destroy
This will delete everything, including the lambda url, the docker container and associated task, as well as all the logs
This command is idempotent and safe to run multiple times.
Troubleshooting
Container uses up too much memory and gets killed
By default, Numerai compute nodes are limited to 8GB of RAM. If you need more, you can open up .numerai/variables.tf
and update the fargate_cpu
and fargate_memory
settings to the following:
variable "fargate_cpu" {
description = "Fargate instance CPU units to provision (1 vCPU = 1024 CPU units)"
default = "4096"
}
variable "fargate_memory" {
description = "Fargate instance memory to provision (in MiB)"
default = "30720"
}
30720MB=30GB and is the maximum that Amazon can support.
After you've done this, re-run numerai setup
.
Note: this will raise the costs of running your compute node, see http://fargate-pricing-calculator.site.s3-website-us-east-1.amazonaws.com/ for estimated costs. You only pay for the time it's running, rounded to the nearest minute.
Billing Alerts
Unfortunately, there's no automated way to setup billing alerts. If you wish to be alerted when costs pass some threshold, you should follow the instructions at https://www.cloudberrylab.com/resources/blog/how-to-configure-billing-alerts-for-your-aws-account/ to setup one.
We estimate costs to be less than $5 per month unless your compute takes more than 12 hours a week. Also keep in mind that increasing the RAM/CPU as described in the previous section will increase your costs.
Prerequisites Help
AWS
You need to signup for AWS and create an administrative IAM user
- Sign up for an AWS account: https://portal.aws.amazon.com/billing/signup
- Create an IAM user with Administrative access: https://console.aws.amazon.com/iam/home?region=us-east-1#/users$new
- Give user a name and select "Programmatic access"
- For permissions, click "Attach existing policies directly" and click the check box next to "AdministratorAccess"
- Save the "Access key ID" and "Secret access key" from the last step. You will need them later
Numerai API Key
- You will need to create an API key by going to https://numer.ai/account and clicking "Add" under the "Your API keys" section.
- Select the following permissions for the key: "Upload submissions", "Make stakes", "View historical submission info", "View user info"
- Your secret key will pop up in the bottom left of the page. Copy this somewhere safe.
- You public ID will be listed when you click "View" under "Your API keys". Copy this somewhere safe as well.
Python
If you don't already have Python3, you can get it from https://www.python.org/downloads/ or install it from your system's package manager.
conda is also a good option to get a working Python environment out of the box
Docker
MacOS and Windows
Just install Docker Desktop to get it running:
https://www.docker.com/products/docker-desktop
You should also increase the RAM allocated to the VM by opening the Docker Desktop GUI then: 1. Click Gear in top right corner 2. Select Resources > Advanced in left sidebar 3. Use slider to allocate more memory (leave a few gigs for your OS and background applications, otherwise your computer might crash)
Docker Toolbox
If your machine doesn't have Hyper-V enabled, then you will have to install docker toolbox: https://github.com/docker/toolbox/releases
After it's installed, open the "Docker QuickStart Terminal" and run the following to increase its RAM:
docker-machine rm default
docker-machine create -d virtualbox --virtualbox-cpu-count=2 --virtualbox-memory=4096 --virtualbox-disk-size=50000 default
Also note, your code must live somewhere under your User directory (ie. C:\Users\USER_NAME\ANY_FOLDER). This is a restriction of docker toolbox not sharing paths correctly otherwise.
Linux
Check here for instructions for your distro: https://docs.docker.com/engine/install/
Uninstall
Destroy the AWS environment
numerai destroy
And then uninstall the package:
pip3 uninstall numerai-cli
Contributions
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
File details
Details for the file numerai-cli-0.2.0.tar.gz
.
File metadata
- Download URL: numerai-cli-0.2.0.tar.gz
- Upload date:
- Size: 5.4 MB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/3.3.0 pkginfo/1.6.1 requests/2.25.1 setuptools/51.1.2 requests-toolbelt/0.9.1 tqdm/4.56.0 CPython/3.7.9
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 1143899e6347e0ea1938f473744b56e8d0392b8b8b7dbe60f89d3281cd0fd34a |
|
MD5 | 1418eede3844fd08550c88452164c72a |
|
BLAKE2b-256 | 048ae2b81a2cf9f53029d6df1873c01082b28052544d262612b27eadb1695b97 |
Provenance
File details
Details for the file numerai_cli-0.2.0-py3-none-any.whl
.
File metadata
- Download URL: numerai_cli-0.2.0-py3-none-any.whl
- Upload date:
- Size: 5.4 MB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/3.3.0 pkginfo/1.6.1 requests/2.25.1 setuptools/51.1.2 requests-toolbelt/0.9.1 tqdm/4.56.0 CPython/3.7.9
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 20e713a46965094e1aa597941efa3950787d4e365e61c307433241160f2f0c8d |
|
MD5 | db3c2e5179dd52fcb0628c3ddcd84bd2 |
|
BLAKE2b-256 | 2b2fc5307bcadab9bff8a404554314c70547b3d4f20e35606e5a96fea95e8361 |