diff --git a/.travis.yml b/.travis.yml
index 986a81fea98bd35c451be7d4824fb128fef9f897..57e9ca410236a472c03f820d148bab3241e05a94 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -7,7 +7,7 @@ language: python
 virtualenv:
   system_site_packages: false
 before_install:
-  - sudo apt-get install -y ninja-build build-essential
+  - sudo apt-get install -y ninja-build build-essential nvidia-cuda-toolkit
 addons:
   apt:
     update: true