prop envs

.travis.yml

@@ -18,7 +18,7 @@ env:
     - FORCE_CUDA=1 CUDA=10.1.105-1 CUDA_SHORT=10.1 UBUNTU_VERSION=ubuntu1804 CUBLAS=libcublas-dev TORCH=${TORCH_VERSION}
 
 before_install:
-  - bash install.sh
+  - source install.sh
 
 install:
   - pip install numpy

install.sh

 #!/bin/bash
 
-if [ "$TRAVIS_OS_NAME" = "linux" ] && [ -z "${CUDA}" ]; then
-  export INSTALLER=cuda-repo-${UBUNTU_VERSION}_${CUDA}_amd64.deb
+if [ "$TRAVIS_OS_NAME" = "linux" ] && [ "${FORCE_CUDA}" != "1" ]; then
+  INSTALLER=cuda-repo-${UBUNTU_VERSION}_${CUDA}_amd64.deb
   wget "http://developer.download.nvidia.com/compute/cuda/repos/${UBUNTU_VERSION}/x86_64/${INSTALLER}"
   sudo dpkg -i "${INSTALLER}"
   wget "https://developer.download.nvidia.com/compute/cuda/repos/${UBUNTU_VERSION}/x86_64/7fa2af80.pub"
@@ -9,8 +9,8 @@ if [ "$TRAVIS_OS_NAME" = "linux" ] && [ -z "${CUDA}" ]; then
   sudo apt update -qq
   sudo apt install -y "cuda-core-${CUDA_SHORT/./-}" "cuda-cudart-dev-${CUDA_SHORT/./-}" "${CUBLAS}" "cuda-cusparse-dev-${CUDA_SHORT/./-}"
   sudo apt clean
-  export CUDA_HOME=/usr/local/cuda-${CUDA_SHORT}
-  export LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
-  export PATH=${CUDA_HOME}/bin:${PATH}
+  CUDA_HOME=/usr/local/cuda-${CUDA_SHORT}
+  LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
+  PATH=${CUDA_HOME}/bin:${PATH}
   nvcc --version
 fi