diff --git a/.drone.yml b/.drone.yml
index b2fa9776..a321fff2 100644
--- a/.drone.yml
+++ b/.drone.yml
@@ -41,7 +41,7 @@ steps:
   commands:
   - pip install -e .[all]
   - pip uninstall -y torchvision
-  - pip install https://download.pytorch.org/whl/cpu/torchvision-0.3.0-cp36-cp36m-linux_x86_64.whl
+  - pip install https://download.pytorch.org/whl/cu100/torchvision-0.3.0-cp36-cp36m-linux_x86_64.whl
   - "python -c 'import torchvision'"
   - python -m unittest tests/*.py