Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
torch-scatter
Commits
a3df06a6
Commit
a3df06a6
authored
Apr 22, 2020
by
rusty1s
Browse files
update to cuda 10.2
parent
dc746d85
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
25 additions
and
32 deletions
+25
-32
.travis.yml
.travis.yml
+10
-18
script/cuda.sh
script/cuda.sh
+15
-14
No files found.
.travis.yml
View file @
a3df06a6
...
...
@@ -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=cu
9
2
env
:
TORCH_VERSION=1.5.0 PYTHON_VERSION=3.
8
IDX=cu
10
2
-
os
:
osx
env
:
TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cu
100
env
:
TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cu
92
-
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=cu
9
2
env
:
TORCH_VERSION=1.5.0 PYTHON_VERSION=3.
7
IDX=cu
10
2
-
os
:
osx
env
:
TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cu
100
env
:
TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cu
92
-
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
...
...
script/cuda.sh
View file @
a3df06a6
...
...
@@ -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
"
=
"cu10
0
"
]
;
then
export
CUDA_SHORT
=
10.
0
if
[
"
${
TRAVIS_OS_NAME
}
"
=
"windows"
]
&&
[
"
$IDX
"
=
"cu10
1
"
]
;
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
}
.1
3
0_41
1.31
_win10
export
CUDA_FILE
=
cuda_
${
CUDA_SHORT
}
.10
5
_41
8.96
_win10
.exe
export
TOOLKIT
=
"cudatoolkit=
${
CUDA_SHORT
}
"
fi
if
[
"
${
TRAVIS_OS_NAME
}
"
=
"windows"
]
&&
[
"
$IDX
"
=
"cu10
1
"
]
;
then
export
CUDA_SHORT
=
10.
1
if
[
"
${
TRAVIS_OS_NAME
}
"
=
"windows"
]
&&
[
"
$IDX
"
=
"cu10
2
"
]
;
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
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment