Skip to content
Snippets Groups Projects

final pytorch install missed

parent 8e100526
No related branches found
No related tags found
No related merge requests found
Pipeline #59664 passed
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment