feat: minimize image size (#5)

* feat: minimize image size
superlinear-ai · Sep 4, 2023 · 0bcbfd9 · 0bcbfd9
1 parent f9a50ae
commit 0bcbfd9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -32,4 +32,4 @@ jobs:
         tags: radixai/python-gpu:${{ matrix.python-version }}-cuda${{ matrix.cuda-version }}
 
     - name: Test Docker image
-      run: docker run --rm radixai/python-gpu:${{ matrix.python-version }}-cuda${{ matrix.cuda-version }} nvcc --version
+      run: docker run --rm radixai/python-gpu:${{ matrix.python-version }}-cuda${{ matrix.cuda-version }}
diff --git a/Dockerfile b/Dockerfile
@@ -16,4 +16,4 @@ RUN CONDA_ARCH=$([ "$TARGETARCH" = "arm64" ] && echo "aarch64" || echo "x86_64")
 ENV PATH=/opt/conda/bin:$PATH
 
 # Install CUDA.
-RUN conda install --channel nvidia --yes cuda="$CUDA_VERSION"
+RUN conda install --channel nvidia --yes cuda-runtime="$CUDA_VERSION"