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
704f831b
"docs/zh_cn/git@developer.sourcefind.cn:OpenDAS/mmcv.git" did not exist on "6f674c7ea6a307a62c0ea74cc60e55570de31f6b"
Unverified
Commit
704f831b
authored
Sep 01, 2023
by
Andrey Talman
Committed by
GitHub
Sep 01, 2023
Browse files
Use regular env setup (#7924)
parent
1044d0fb
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
15 additions
and
5 deletions
+15
-5
.github/scripts/setup-env.sh
.github/scripts/setup-env.sh
+15
-5
No files found.
.github/scripts/setup-env.sh
View file @
704f831b
...
...
@@ -73,11 +73,21 @@ else
CHANNEL
=
nightly
fi
pip
install
--progress-bar
=
off light-the-torch
ltt
install
--progress-bar
=
off
\
--pytorch-computation-backend
=
"
${
GPU_ARCH_TYPE
}${
GPU_ARCH_VERSION
}
"
\
--pytorch-channel
=
"
${
CHANNEL
}
"
\
torch
case
$GPU_ARCH_TYPE
in
cpu
)
GPU_ARCH_ID
=
"cpu"
;;
cuda
)
VERSION_WITHOUT_DOT
=
$(
echo
"
${
GPU_ARCH_VERSION
}
"
|
sed
's/\.//'
)
GPU_ARCH_ID
=
"cu
${
VERSION_WITHOUT_DOT
}
"
;;
*
)
echo
"Unknown GPU_ARCH_TYPE=
${
GPU_ARCH_TYPE
}
"
exit
1
;;
esac
PYTORCH_WHEEL_INDEX
=
"https://download.pytorch.org/whl/
${
CHANNEL
}
/
${
GPU_ARCH_ID
}
"
pip
install
--progress-bar
=
off
--pre
torch
--index-url
=
"
${
PYTORCH_WHEEL_INDEX
}
"
if
[[
$GPU_ARCH_TYPE
==
'cuda'
]]
;
then
python
-c
"import torch; exit(not torch.cuda.is_available())"
...
...
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