From 4311ba0653b79c8ef6a6ac2e2af2a02fd66a6bed Mon Sep 17 00:00:00 2001
From: ct00659um <ct00659@surrey.ac.uk>
Date: Thu, 9 Nov 2023 18:36:37 +0000
Subject: [PATCH] New cuda version
---
Dockerfile | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/Dockerfile b/Dockerfile
index 488800f..317f50d 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
--
GitLab