From 38822058903f20e7839a29b81a025ac022d71e20 Mon Sep 17 00:00:00 2001
From: ct00659um <ct00659@surrey.ac.uk>
Date: Thu, 9 Nov 2023 17:48:04 +0000
Subject: [PATCH] final pytorch install missed
---
Dockerfile | 7 ++++++-
1 file changed, 6 insertions(+), 1 deletion(-)
diff --git a/Dockerfile b/Dockerfile
index aa01174..488800f 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -51,7 +51,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 . && \
- pip install opencv-python pycocotools matplotlib onnxruntime onnx
+ pip install opencv-python pycocotools matplotlib onnxruntime onnx && \
+ conda install pytorch torchvision torchaudio cudatoolkit=11.2 -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
@@ -59,6 +60,10 @@ RUN pip install setuptools==60.9.3 && \
# Set the default command to bash to activate conda environment on run
CMD ["/bin/bash"]
+# RUN conda install pytorch torchvision torchaudio cudatoolkit=11.2 -c pytorch -c nvidia
+
+# RUN conda activate segment_anything_env
+
# RUN conda install pytorch>=1.7 torchvision>=0.8 torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
#RUN pip3 install ffmpeg gnutls
--
GitLab