Skip to content
Snippets Groups Projects
Commit 4311ba06 authored by Thirgood, Christopher T (PG/R - Comp Sci & Elec Eng)'s avatar Thirgood, Christopher T (PG/R - Comp Sci & Elec Eng)
Browse files

New cuda version

parent 38822058
No related branches found
No related tags found
No related merge requests found
Pipeline #59670 failed
Stage: build
Stage: push
Hide whitespace changes
Inline Side-by-side
Showing
with 3 additions and 2 deletions
Dockerfile
+ 3
− 2
View file @ 4311ba06
FROM nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04 FROM nvidia/cuda:11.2.2-cudnn8-devel-ubuntu22.04
ARG DEBIAN_FRONTEND=noninteractive ARG DEBIAN_FRONTEND=noninteractive
...@@ -52,7 +52,8 @@ RUN pip install setuptools==60.9.3 && \ ...@@ -52,7 +52,8 @@ RUN pip install setuptools==60.9.3 && \
git clone https://github.com/facebookresearch/segment-anything.git && \ git clone https://github.com/facebookresearch/segment-anything.git && \
cd segment-anything && pip install -e . && \ 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 # 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 # 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 # RUN conda install pytorch torchvision torchaudio cudatoolkit=11.2 -c pytorch -c nvidia
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment