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
vision
Commits
00670edc
Commit
00670edc
authored
Jun 07, 2023
by
panning
Browse files
update dcu_version
parent
5343fb43
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
setup.py
setup.py
+2
-2
No files found.
setup.py
View file @
00670edc
...
@@ -74,7 +74,7 @@ def write_version_file():
...
@@ -74,7 +74,7 @@ def write_version_file():
version_path
=
os
.
path
.
join
(
cwd
,
"torchvision"
,
"version.py"
)
version_path
=
os
.
path
.
join
(
cwd
,
"torchvision"
,
"version.py"
)
with
open
(
version_path
,
"w"
)
as
f
:
with
open
(
version_path
,
"w"
)
as
f
:
f
.
write
(
f
"__version__ = '
{
version
}
'
\n
"
)
f
.
write
(
f
"__version__ = '
{
version
}
'
\n
"
)
f
.
write
(
"__dcu_version__ = '{
}'
\n
"
.
format
(
dcu_version
)
)
f
.
write
(
f
"__dcu_version__ = '
{
dcu_version
}
'
\n
"
)
f
.
write
(
f
"git_version =
{
repr
(
sha
)
}
\n
"
)
f
.
write
(
f
"git_version =
{
repr
(
sha
)
}
\n
"
)
f
.
write
(
"from torchvision.extension import _check_cuda_version
\n
"
)
f
.
write
(
"from torchvision.extension import _check_cuda_version
\n
"
)
f
.
write
(
"if _check_cuda_version() > 0:
\n
"
)
f
.
write
(
"if _check_cuda_version() > 0:
\n
"
)
...
@@ -552,7 +552,7 @@ if __name__ == "__main__":
...
@@ -552,7 +552,7 @@ if __name__ == "__main__":
write_version_file
()
write_version_file
()
with
open
(
"README.rst"
)
as
f
:
with
open
(
"README
_ORIGIN
.rst"
)
as
f
:
readme
=
f
.
read
()
readme
=
f
.
read
()
setup
(
setup
(
...
...
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