feat: minimize image size (#5)

* feat: minimize image size

Author: lsorber

.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 }}

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"