muryshev commited on
Commit
485eb7b
·
verified ·
1 Parent(s): b1cce5f

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -1,5 +1,5 @@
1
  ARG UBUNTU_VERSION=22.04
2
- ARG CUDA_VERSION=12.3.1
3
  ARG BASE_CUDA_DEV_CONTAINER=nvidia/cuda:${CUDA_VERSION}-devel-ubuntu${UBUNTU_VERSION}
4
  ARG BASE_CUDA_RUN_CONTAINER=nvidia/cuda:${CUDA_VERSION}-runtime-ubuntu${UBUNTU_VERSION}
5
 
@@ -8,7 +8,7 @@ FROM ${BASE_CUDA_DEV_CONTAINER} as build
8
  ARG CUDA_DOCKER_ARCH="default"
9
 
10
  RUN apt-get update --fix-missing && \
11
- apt-get install -y --no-install-recommends git build-essential gcc cmake && \
12
  rm -rf /var/lib/apt/lists/*
13
 
14
  WORKDIR /build
 
1
  ARG UBUNTU_VERSION=22.04
2
+ ARG CUDA_VERSION=12.6.0
3
  ARG BASE_CUDA_DEV_CONTAINER=nvidia/cuda:${CUDA_VERSION}-devel-ubuntu${UBUNTU_VERSION}
4
  ARG BASE_CUDA_RUN_CONTAINER=nvidia/cuda:${CUDA_VERSION}-runtime-ubuntu${UBUNTU_VERSION}
5
 
 
8
  ARG CUDA_DOCKER_ARCH="default"
9
 
10
  RUN apt-get update --fix-missing && \
11
+ apt-get install -y --no-install-recommends git build-essential gcc cmake curl && \
12
  rm -rf /var/lib/apt/lists/*
13
 
14
  WORKDIR /build