env.sh

#!/bin/bash

export TORCH_CUDA_ARCH_LIST="{archs}"
export IABN_FORCE_CUDA=1