Skip to content

Commit ccb29b2

Browse files
committed
fix: Pytests
1 parent 9bfd582 commit ccb29b2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

label_studio_ml/examples/yolo/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ ENV PYTHONUNBUFFERED=1 \
2323
THREADS=4 \
2424
CUDA_HOME=/usr/local/cuda
2525

26-
RUN conda install -c "nvidia/label/cuda-12.1.1" cuda -y
26+
# RUN conda install -c "nvidia/label/cuda-12.1.1" cuda -y
2727
ENV CUDA_HOME=/opt/conda \
2828
TORCH_CUDA_ARCH_LIST="6.0;6.1;7.0;7.5;8.0;8.6+PTX;8.9;9.0"
2929

0 commit comments

Comments
 (0)