CUDA MPS start and stop script

start_mps_as_root.sh

#!/bin/bash
# the following must be performed with root privilege

helpFuction()
{
	echo ""
	echo "Usage: $0 device_id"
	echo -e "\t device_id is the id of nvidia device from what nvidia-smi displays"
	exit 1
}

if [ -z $1 ];
then
	helpFuction
fi

echo "Set Device $1"

export CUDA_VISIBLE_DEVICES="$1"
nvidia-smi -i "$1" -c EXCLUSIVE_PROCESS
nvidia-cuda-mps-control -d

stop_mps_as_root.sh

#!/bin/bash
# the following must be performed with root privilege

helpFuction()
{
	echo ""
	echo "Usage: $0 device_id"
	echo -e "\t device_id is the id of nvidia device from what nvidia-smi displays"
	exit 1
}

if [ -z $1 ];
then
	helpFuction
fi

echo "Set Device $1"
echo quit | nvidia-cuda-mps-control
nvidia-smi -i "$1" -c DEFAULT