diff --git a/Dockerfile b/Dockerfile index 68f16c158c159a312f0c4379963afb463b01871b..12b3a7e18a7095d23487aef69c77aaa510b15289 100644 --- a/Dockerfile +++ b/Dockerfile @@ -67,8 +67,8 @@ SHELL ["conda", "run", "-n", "segment_anything_env", "/bin/bash", "-c"] RUN pip install setuptools==60.9.3 && \ git clone https://github.com/facebookresearch/segment-anything.git && \ cd segment-anything && pip install -e . && \ - conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch && \ - pip install opencv-python pycocotools matplotlib onnxruntime onnx + conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch +#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