File tree Expand file tree Collapse file tree 2 files changed +6
-2
lines changed
Expand file tree Collapse file tree 2 files changed +6
-2
lines changed Original file line number Diff line number Diff line change @@ -24,7 +24,9 @@ ENV PYTHONUNBUFFERED=1 \
2424 CUDA_HOME=/usr/local/cuda \
2525 SEGMENT_ANYTHING_2_REPO_PATH=/segment-anything-2
2626
27- RUN conda install -c "nvidia/label/cuda-12.1.1" cuda -y
27+ # this image is already built with cuda 12.1.1, so we don't need to install it again
28+ # RUN conda install -c "nvidia/label/cuda-12.1.1" cuda -y
29+
2830ENV CUDA_HOME=/opt/conda \
2931 TORCH_CUDA_ARCH_LIST="6.0;6.1;7.0;7.5;8.0;8.6+PTX;8.9;9.0"
3032
Original file line number Diff line number Diff line change @@ -24,7 +24,9 @@ ENV PYTHONUNBUFFERED=1 \
2424 CUDA_HOME=/usr/local/cuda \
2525 SEGMENT_ANYTHING_2_REPO_PATH=/segment-anything-2
2626
27- RUN conda install -c "nvidia/label/cuda-12.1.1" cuda -y
27+ # this image is already built with cuda 12.1.1, so we don't need to install it again
28+ # RUN conda install -c "nvidia/label/cuda-12.1.1" cuda -y
29+
2830ENV CUDA_HOME=/opt/conda \
2931 TORCH_CUDA_ARCH_LIST="6.0;6.1;7.0;7.5;8.0;8.6+PTX;8.9;9.0"
3032
You can’t perform that action at this time.
0 commit comments