Unable to build a new image from rapidsai:cuda10.2-runtime-ubuntu18.04 with additional python libraries

Accelerated Computing NGC GPU Cloud Docker and NVIDIA Docker
1

I am using the following Dockerfile to build a new image

FROM nvcr.io/nvidia/rapidsai/rapidsai:cuda10.2-runtime-ubuntu18.04

WORKDIR /rapids/notebooks/

COPY requirements.txt ./

RUN pip install ujson

The Dockerfile runs without any error but I cannot import ujson from the newly build container. The difference is that in the new container usjon is installed for cp38. If I do pip install inside the newly built container it would be for cp37.

Trace when running pip install inside the container

Collecting ujson
  Downloading ujson-4.0.1-cp37-cp37m-manylinux1_x86_64.whl (179 kB)
     |████████████████████████████████| 179 kB 947 kB/s
Installing collected packages: ujson
Successfully installed ujson-4.0.1

Trace when running Dockerfile

Collecting ujson
  Downloading ujson-4.0.1-cp38-cp38-manylinux1_x86_64.whl (181 kB)
Installing collected packages: ujson
Successfully installed ujson-4.0.1

can somebody help me understand what’s the issue, How can I solve for this?