security_vision/alphapose-docker
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

fix default python command

Browse source
This commit is contained in:
Ruben van de Ven 2023-01-19 11:21:07 +01:00
parent fa5871bae9
commit bbb0518d29
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
Dockerfile
View file

@ -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