From 0c2be0361abd09d721fd22e4c1135fcc2b7c3d8d Mon Sep 17 00:00:00 2001
From: Meng Zhang <meng@tabbyml.com>
Date: Sun, 19 Nov 2023 14:08:57 -0800
Subject: [PATCH] fix: RUST_TOOLCHAIN need to be inside of from block (#840)

---
 Dockerfile | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 48dca6c..69ac755 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -5,11 +5,12 @@ ARG CUDA_VERSION=11.7.1
 ARG BASE_CUDA_DEV_CONTAINER=nvidia/cuda:${CUDA_VERSION}-devel-ubuntu${UBUNTU_VERSION}
 # Target the CUDA runtime image
 ARG BASE_CUDA_RUN_CONTAINER=nvidia/cuda:${CUDA_VERSION}-runtime-ubuntu${UBUNTU_VERSION}
-# Rust toolchain version
-ARG RUST_TOOLCHAIN stable
 
 FROM ${BASE_CUDA_DEV_CONTAINER} as build
 
+# Rust toolchain version
+ARG RUST_TOOLCHAIN stable
+
 ENV DEBIAN_FRONTEND=noninteractive
 RUN apt-get update && \
     apt-get install -y --no-install-recommends \