We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9bfd582 commit ccb29b2Copy full SHA for ccb29b2
label_studio_ml/examples/yolo/Dockerfile
@@ -23,7 +23,7 @@ ENV PYTHONUNBUFFERED=1 \
23
THREADS=4 \
24
CUDA_HOME=/usr/local/cuda
25
26
-RUN conda install -c "nvidia/label/cuda-12.1.1" cuda -y
+# RUN conda install -c "nvidia/label/cuda-12.1.1" cuda -y
27
ENV CUDA_HOME=/opt/conda \
28
TORCH_CUDA_ARCH_LIST="6.0;6.1;7.0;7.5;8.0;8.6+PTX;8.9;9.0"
29
0 commit comments