diff --git a/Dockerfile b/Dockerfile index 1c59f209f0c19d697b2d91ca765f57ab54eaa86a..cc4f92f38f3f0164742bf9c73ef046f01633da16 100644 --- a/Dockerfile +++ b/Dockerfile @@ -68,7 +68,9 @@ RUN pip install setuptools==60.9.3 && \ git clone https://github.com/facebookresearch/segment-anything.git && \ cd segment-anything && pip install -e . && \ pip install opencv-python pycocotools matplotlib onnxruntime onnx && \ - conda install pytorch torchvision torchaudio cudatoolkit=11.2 -c pytorch -c nvidia + conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch +#conda install pytorch torchvision torchaudio cudatoolkit=11.2 -c pytorch -c nvidia + #conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch # Set the default command to bash to activate conda environment on run CMD ["/bin/bash"]