ucx_playground

Pre-requisites

Build and install UCX locally. Refer to the UCX Github releases. Pick a release that you prefer and build locally. Then include it to the LD_LIBRARY_PATH

UCX Documentation

export UCX_HOME=<your-path-to-ucx>
export LD_LIBRARY_PATH=$UCX_HOME/build/lib${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export PATH=$PATH:$UCX_HOME/build/bin

Create Environment

mamba create -n ucx_env python=3.9
mamba activate ucx_env
mamba install pre-commit
pre-commit install
pre-commit install --hook-type pre-push

Build Project

cd cpp
mkdir build
cd build
cmake ../
make

Run Sample App

cd cpp/build
./main

Run UCX Hello World By NVidia

This example has been adopted from examples in UCX Github.

Terminal 1: Running server

cd cpp/build
./hello_ucp

Terminal 2: Running client

cd cpp/build
./hello_ucp -n 0.0.0.0

Server/Client Program

./run_ucp_server

./run_ucp_client -n 0.0.0.0