Commit a3df06a6 authored by rusty1s's avatar rusty1s
Browse files

update to cuda 10.2

parent dc746d85
......@@ -11,51 +11,43 @@ env:
jobs:
- TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cpu
- TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cu92
- TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cu100
- TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cu101
- TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cu102
- TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cpu
- TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cu92
- TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cu100
- TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cu101
- TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cu102
- TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cpu
- TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cu92
- TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cu100
- TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cu101
- TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cu102
jobs:
exclude: # Exclude *all* macOS CUDA jobs and Windows CUDA 9.2/10.0 jobs.
exclude: # Exclude *all* macOS CUDA jobs and Windows CUDA 9.2 jobs.
- os: osx
env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cu92
- os: osx
env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cu100
- os: osx
env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cu101
- os: osx
env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cu92
env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cu102
- os: osx
env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cu100
env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cu92
- os: osx
env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cu101
- os: osx
env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cu92
env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cu102
- os: osx
env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cu100
env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cu92
- os: osx
env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cu101
- os: osx
env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cu102
- os: windows
env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cu92
- os: windows
env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cu100
- os: windows
env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cu92
- os: windows
env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cu100
- os: windows
env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cu92
- os: windows
env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cu100
- os: windows
env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cu101
install:
- source script/cuda.sh
......
......@@ -12,14 +12,6 @@ if [ "${TRAVIS_OS_NAME}" = "linux" ] && [ "$IDX" = "cu92" ]; then
export TOOLKIT="cudatoolkit=${CUDA_SHORT}"
fi
if [ "${TRAVIS_OS_NAME}" = "linux" ] && [ "$IDX" = "cu100" ]; then
export CUDA_SHORT=10.0
export CUDA=10.0.130-1
export UBUNTU_VERSION=ubuntu1804
export CUBLAS=cuda-cublas-dev-10-0
export TOOLKIT="cudatoolkit=${CUDA_SHORT}"
fi
if [ "${TRAVIS_OS_NAME}" = "linux" ] && [ "$IDX" = "cu101" ]; then
export IDX=cu101
export CUDA_SHORT=10.1
......@@ -29,6 +21,15 @@ if [ "${TRAVIS_OS_NAME}" = "linux" ] && [ "$IDX" = "cu101" ]; then
export TOOLKIT="cudatoolkit=${CUDA_SHORT}"
fi
if [ "${TRAVIS_OS_NAME}" = "linux" ] && [ "$IDX" = "cu102" ]; then
export CUDA_SHORT=10.2
export CUDA=10.2.89-1
export UBUNTU_VERSION=ubuntu1804
export CUBLAS=libcublas-dev
export TOOLKIT="cudatoolkit=${CUDA_SHORT}"
fi
if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "$IDX" = "cpu" ]; then
export TOOLKIT=cpuonly
fi
......@@ -40,17 +41,17 @@ if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "$IDX" = "cu92" ]; then
export TOOLKIT="cudatoolkit=${CUDA_SHORT}"
fi
if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "$IDX" = "cu100" ]; then
export CUDA_SHORT=10.0
if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "$IDX" = "cu101" ]; then
export CUDA_SHORT=10.1
export CUDA_URL=https://developer.nvidia.com/compute/cuda/${CUDA_SHORT}/Prod/local_installers
export CUDA_FILE=cuda_${CUDA_SHORT}.130_411.31_win10
export CUDA_FILE=cuda_${CUDA_SHORT}.105_418.96_win10.exe
export TOOLKIT="cudatoolkit=${CUDA_SHORT}"
fi
if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "$IDX" = "cu101" ]; then
export CUDA_SHORT=10.1
if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "$IDX" = "cu102" ]; then
export CUDA_SHORT=10.2
export CUDA_URL=https://developer.nvidia.com/compute/cuda/${CUDA_SHORT}/Prod/local_installers
export CUDA_FILE=cuda_${CUDA_SHORT}.105_418.96_win10.exe
export CUDA_FILE=cuda_${CUDA_SHORT}.89_441.22_win10
export TOOLKIT="cudatoolkit=${CUDA_SHORT}"
fi
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment