cuda and pytorch version when building docker

Hello. I used mmcv to construct my model and it requires complilation before use. I have to know the cuda and pytorch version before I can choose the right version of mmcv to build. So if it is possible to know the cuda version on the machine where our docker will be built? The pytorch version I assume will be the latest as it implied in the dockerfile? Thank you.

Hi,

You can specify which pytorch version and cuda version you want to use by modifying the first line in your Dockerfile. For example: FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime for pytorch version 1.10.0, and cuda 11.3. Here is a list of base images from pytorch/pytorch that you could build upon: https://hub.docker.com/r/pytorch/pytorch/tags

Kind regards, Luuk

Thank you a lot.