Pulling the docker image

```bash
docker pull nvcr.io/nvidia/pytorch:23.09-py3
```

Runing the docker with

   docker run --gpus all -it --rm -v ./:/data nvcr.io/nvidia/pytorch:23.09-py3

Importing ASP to test it

   python
   from apex.contrib.sparsity import ASP