Skip to content

Commit 30ed610

Browse files
committed
Add install of torch.
1 parent c5951f8 commit 30ed610

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tools/ci/check_validate.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@ function prepare_env() {
2626
env http_proxy="" https_proxy="" pip install -U pip > /dev/null
2727
[ $? -ne 0 ] && LOG "[FATAL] Update pip failed!" && exit -1
2828
# install torch
29+
pip install torch==2.7.0 --index-url https://download.pytorch.org/whl/cu118
2930
}
3031

3132
function check_validation() {

0 commit comments

Comments
 (0)