Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
one
spconv
Commits
3c1f4198
"csrc/extension/softmax.h" did not exist on "5813dcc146f05651c23536269397a12844841b9b"
Commit
3c1f4198
authored
Oct 21, 2021
by
Yan Yan
Browse files
finally, we find the windows CUDA bug
parent
2b511849
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
30 additions
and
17 deletions
+30
-17
.github/workflows/build.yaml
.github/workflows/build.yaml
+1
-10
tools/install_windows_cuda.ps1
tools/install_windows_cuda.ps1
+29
-7
No files found.
.github/workflows/build.yaml
View file @
3c1f4198
...
...
@@ -12,7 +12,7 @@ on:
jobs
:
build-windows
:
runs-on
:
windows-
latest
runs-on
:
windows-
2019
strategy
:
matrix
:
python-version
:
[
'
3.7'
,
'
3.8'
,
'
3.9'
,
'
3.10'
]
...
...
@@ -48,16 +48,7 @@ jobs:
$Env:CUMM_CUDA_ARCH_LIST = "all"
$Env:SPCONV_DISABLE_JIT = "1"
pip install pccm pybind11
$CUDA_INCLUDE_THRUST = "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v${{ matrix.cuda-version }}\include\thrust"
if (Test-Path -Path $CUDA_INCLUDE_THRUST) {
} else {
git clone -b cuda-${{ matrix.cuda-version }} https://github.com/NVIDIA/thrust $HOME/thrust_git
ls $HOME/thrust_git/thrust
mv $HOME/thrust_git/thrust "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v${{ matrix.cuda-version }}\include"
}
ls "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v${{ matrix.cuda-version }}\include\thrust"
$Env:CUMM_THRUST_INCLUDE = "$HOME/thrust_git"
python -m build --wheel --outdir dist/ .
shell
:
powershell
...
...
tools/install_windows_cuda.ps1
View file @
3c1f4198
...
...
@@ -10,17 +10,38 @@ $CUDA_KNOWN_URLS = @{
"11.2"
=
"https://developer.download.nvidia.com/compute/cuda/11.2.2/network_installers/cuda_11.2.2_win10_network.exe"
;
"11.3"
=
"https://developer.download.nvidia.com/compute/cuda/11.3.1/network_installers/cuda_11.3.1_win10_network.exe"
;
"11.4"
=
"https://developer.download.nvidia.com/compute/cuda/11.4.2/network_installers/cuda_11.4.2_win10_network.exe"
;
"11.5"
=
"https://developer.download.nvidia.com/compute/cuda/11.5.0/network_installers/cuda_11.5.0_win10_network.exe"
;
}
# cuda_runtime.h is in nvcc <= 10.2, but cudart >= 11.0
# @todo - make this easier to vary per CUDA version.
$CUDA_PACKAGES_IN
=
@(
if
((
$CUDA_VERSION_FULL
-eq
"10.2"
)
-or
(
$CUDA_VERSION_FULL
-eq
"11.0"
)
-or
(
$CUDA_VERSION_FULL
-eq
"11.1"
)
-or
(
$CUDA_VERSION_FULL
-eq
"11.2"
)){
$CUDA_PACKAGES_IN
=
@(
"cuda_nvcc"
;
"visual_studio_integration"
;
"nvrtc_dev"
;
"cudart"
;
# before 11.3, thrust are included by default and no explicit package exists
)
}
elseif
(
$CUDA_VERSION_FULL
-eq
"11.3"
){
$CUDA_PACKAGES_IN
=
@(
"nvcc"
;
"visual_studio_integration"
;
"cuda_nvrtc"
;
"cuda_cudart"
;
"cuda_thrust"
;
)
}
else
{
# after cuda 11.4
$CUDA_PACKAGES_IN
=
@(
"nvcc"
;
"visual_studio_integration"
;
"curand_dev"
;
"nvrtc_dev"
;
"cudart"
;
)
"thrust"
;
)
}
## -------------------
...
...
@@ -40,6 +61,7 @@ if(-not $cuda_ver_matched){
$CUDA_MAJOR
=
$Matches
.
major
$CUDA_MINOR
=
$Matches
.
minor
## ------------------------------------------------
## Select CUDA packages to install from environment
## ------------------------------------------------
...
...
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