Skip to content

Commit

Permalink
Update Dockerfile
Browse files Browse the repository at this point in the history
  • Loading branch information
@PGerardi
PGerardi authored Sep 7, 2023
1 parent 9fa4260 commit ec4eac7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion Dockerfile
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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"

0 comments on commit ec4eac7

Please sign in to comment.