diff --git a/dockerfile/cuda11.1.1.dockerfile b/dockerfile/cuda11.1.1.dockerfile index 7ee352543..662c36156 100644 --- a/dockerfile/cuda11.1.1.dockerfile +++ b/dockerfile/cuda11.1.1.dockerfile @@ -24,6 +24,7 @@ RUN apt-get update && \ automake \ bc \ build-essential \ + ca-certificates \ curl \ dmidecode \ ffmpeg \ @@ -46,6 +47,7 @@ RUN apt-get update && \ openssh-client \ openssh-server \ pciutils \ + software-properties-common \ sudo \ util-linux \ vim \ @@ -55,6 +57,19 @@ RUN apt-get update && \ apt-get clean && \ rm -rf /var/lib/apt/lists/* /tmp/* /opt/cmake-3.14.6-Linux-x86_64 +# Install Go (used for wandb build) +RUN add-apt-repository -y ppa:longsleep/golang-backports && \ + apt-get update && \ + apt-get install -y golang-1.24-go + +# Install Rust (used for wandb build) +RUN curl https://sh.rustup.rs -sSf | sh -s -- -y && \ + . /root/.cargo/env && \ + rustup update stable && \ + rustup default stable + +ENV PATH="/usr/lib/go-1.24/bin:/root/.cargo/bin:${PATH}" + ARG NUM_MAKE_JOBS= # Install Docker diff --git a/superbench/benchmarks/base.py b/superbench/benchmarks/base.py index 8e6e58bfe..46886b4e7 100644 --- a/superbench/benchmarks/base.py +++ b/superbench/benchmarks/base.py @@ -293,10 +293,11 @@ def _process_percentile_result(self, metric, result, reduce_type=None): if len(result) > 0: percentile_list = ['50', '90', '95', '99', '99.9'] for percentile in percentile_list: - self._result.add_result( - '{}_{}'.format(metric, percentile), - np.percentile(result, float(percentile), interpolation='nearest'), reduce_type - ) + try: + val = np.percentile(result, float(percentile), method='nearest') + except TypeError: + val = np.percentile(result, float(percentile), interpolation='nearest') + self._result.add_result('{}_{}'.format(metric, percentile), val, reduce_type) def print_env_info(self): """Print environments or dependencies information."""