travis fix

1e244b3f · rusty1s · 42da9ac2 · 1e244b3f
Commit 1e244b3f authored Apr 23, 2020 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

.travis.yml .travis.yml +4 -0

No files found.
--- a/.travis.yml
+++ b/.travis.yml
@@ -5,6 +5,10 @@ os:
  - osx
  - windows
+env:
+  global:
+    - CUDA_HOME=/usr/local/cuda
  jobs:
    - TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cpu
    - TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cu92