diff --git a/Dockerfile b/Dockerfile index 488800fc7239c0063145c269e0af052021fa146d..317f50d55c86c5283b2cfca1fd68f3d9672af978 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,4 +1,4 @@ -FROM nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04 +FROM nvidia/cuda:11.2.2-cudnn8-devel-ubuntu22.04 ARG DEBIAN_FRONTEND=noninteractive @@ -52,7 +52,8 @@ 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 torchvision torchaudio cudatoolkit=11.2 -c pytorch -c nvidia + conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia # Uncomment the following line if you want to install PyTorch and related packages within the Conda environment # RUN conda install pytorch torchvision torchaudio cudatoolkit=11.2 -c pytorch -c nvidia