Sign inSign up

cainmagi/deformable-detr

By cainmagi

•Updated over 1 year ago

This image is only used for building and testing the usability of Deformable-DETR。

Image
Machine learning & AI
Developer tools
Data science
0

124

cainmagi/deformable-detr repository overview

⁠Cainmagi's Docker images

A collection of my personal customized Docker images.

⁠Deformable-DETR

This image is only used for building and testing the usability of Deformable-DETR⁠.

In this image, we have the following packages installed:

  • PyTorch (torch)
  • Transformers with PyTorch (transformers[torch])
  • MultiScaleDeformableAttention (MultiScaleDeformableAttention)

⁠Usage

Run the following command to pull the image:

docker pull cainmagi/deformable-detr:debian12

After that, run the following command to launch a shell

docker run --gpus all -it --rm --shm-size=1g deformable-detr:debian12 bash

Caution

If you do not add bash at the end of the command, the container may be launched as Python.

⁠More details

To view the scripts used for building these images, please check

https://github.com/cainmagi/Dockerfiles/tree/Deformable-DETR⁠

⁠Tags and explanations

TagOSCUDAPython
debian12Debian 1212.43.12

Tag summary

Content type

Image

Digest

sha256:bc0af0312…

Size

10 GB

Last updated

over 1 year ago

docker pull cainmagi/deformable-detr:debian12