diff --git a/providers/gpgpu/tools/gpu-setup b/providers/gpgpu/tools/gpu-setup
index d896eb971f..971d70c39f 100755
--- a/providers/gpgpu/tools/gpu-setup
+++ b/providers/gpgpu/tools/gpu-setup
@@ -59,7 +59,7 @@ add-apt-repository -y "deb http://developer.download.nvidia.com/compute/cuda/rep
 echo "*  Installing necessary pacakges"
 apt install -y build-essential git
 ## need to break this out to fix issue where cuda installs gdm3
-apt install -y --no-install-recommends cuda
+apt install -y --no-install-recommends cuda-toolkit
 
 #fix the path to get nvcc from the cuda package
 CUDA_PATH=$(find /usr/local -maxdepth 1 -type d -iname "cuda*")/bin
diff --git a/providers/gpgpu/units/jobs.pxu b/providers/gpgpu/units/jobs.pxu
index e65d0be387..83b7050547 100644
--- a/providers/gpgpu/units/jobs.pxu
+++ b/providers/gpgpu/units/jobs.pxu
@@ -3,6 +3,6 @@ category_id: gpgpu
 plugin: shell
 estimated_duration: 300
 requires:
-    package.name == 'cuda'
+    package.name == 'cuda-toolkit'
 _summary: GPGPU stress testing
 command: cd /opt/gpu-burn/ && ./gpu_burn 14400 | grep -v -e '^[[:space:]]*$' -e "errors:" -e "Summary at"