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
deepspeed
Commits
ff58fa7e
Unverified
Commit
ff58fa7e
authored
Dec 02, 2020
by
Stas Bekman
Committed by
GitHub
Dec 02, 2020
Browse files
[build] build against installed cuda-11.1 while torch built w/ cuda-11.0 (#570)
parent
be33bea4
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
0 deletions
+3
-0
op_builder/builder.py
op_builder/builder.py
+3
-0
No files found.
op_builder/builder.py
View file @
ff58fa7e
...
@@ -44,6 +44,9 @@ def assert_no_cuda_mismatch():
...
@@ -44,6 +44,9 @@ def assert_no_cuda_mismatch():
torch_cuda_version
=
"."
.
join
(
torch
.
version
.
cuda
.
split
(
'.'
)[:
2
])
torch_cuda_version
=
"."
.
join
(
torch
.
version
.
cuda
.
split
(
'.'
)[:
2
])
# This is a show-stopping error, should probably not proceed past this
# This is a show-stopping error, should probably not proceed past this
if
sys_cuda_version
!=
torch_cuda_version
:
if
sys_cuda_version
!=
torch_cuda_version
:
if
sys_cuda_version
==
"11.1"
and
torch_cuda_version
==
"11.0"
:
# it works to build against installed cuda-11.1 while torch was built with cuda-11.0
return
raise
Exception
(
raise
Exception
(
f
"Installed CUDA version
{
sys_cuda_version
}
does not match the "
f
"Installed CUDA version
{
sys_cuda_version
}
does not match the "
f
"version torch was compiled with
{
torch
.
version
.
cuda
}
, unable to compile "
f
"version torch was compiled with
{
torch
.
version
.
cuda
}
, unable to compile "
...
...
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