diff --git a/Dockerfile b/Dockerfile index 12625c9..c379522 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,9 +1,12 @@ # thanks to Paul DeCoursey (optimuspaul@github.com) for inspiration # see: https://github.com/MVIG-SJTU/AlphaPose/issues/1057 -#FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04 #FROM nvidia/cuda:11.6.0-cudnn8-devel-ubuntu22.04 +#TODO: fix deprecation with either of these FROM nvidia/cuda:11.6.0-devel-ubuntu20.04 +#FROM nvidia/cuda:11.6.2-devel-ubuntu20.04 +#FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04 +#FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04 RUN apt clean @@ -67,3 +70,4 @@ RUN apt install -y python3-tk # see: https://stackoverflow.com/questions/33935807/how-to-define-a-variable-in-a-dockerfile#37639938 RUN python3 -c "import torchvision.models as tm; tm.resnet50(pretrained=True)" +RUN update-alternatives --install /usr/bin/python python /usr/bin/python3 2