diff --git a/packages/py-vbi/package.py b/packages/py-vbi/package.py
index 601c9bded08c47dc2be8eff33b939acb948199ec..0ac988a2822897fc1d51da03e51f43cd6c15b908 100644
--- a/packages/py-vbi/package.py
+++ b/packages/py-vbi/package.py
@@ -6,7 +6,7 @@
 from spack import *
 
 
-class PyVbi(PythonPackage):
+class PyVbi(PythonPackage, CudaPackage):
 
     homepage = "https://vbi.readthedocs.io/latest/"
     git = "https://github.com/ins-amu/vbi"
@@ -37,7 +37,7 @@ class PyVbi(PythonPackage):
     depends_on("py-scikit-learn", type=("build", "run"))
     depends_on("py-pycatch22", type=("build", "run"))
     depends_on("py-pytest", type="test")
-    depends_on("py-cupy", type=("build", "run"))
+    depends_on("py-cupy", type=("build", "run"), when="+cuda")
     
     @run_after("install")
     @on_package_attributes(run_tests=True)