Skip to content

Commit

Permalink
Update pytorch to 2.4.1 to resovle missing .dll on Windows (#79)
Browse files Browse the repository at this point in the history
  • Loading branch information
maxlou05 committed Sep 5, 2024
1 parent bce5937 commit 96bc51a
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -17,15 +17,14 @@ pytest
# If you are on MacOS, don't have a CUDA capable GPU,
# or if things don't work and you just want to use the CPU,
# comment out the "--extra-index-url" option
--extra-index-url https://download.pytorch.org/whl/cu121
--extra-index-url https://download.pytorch.org/whl/cu124

# ============
# ↑ BOOTCAMPERS MODIFY ABOVE THIS COMMENT ↑
# ============

# TODO: Update to 2.4.1 or later
torch==2.3.1
torchvision==0.18.1
torch==2.4.1
torchvision==0.19.1

ultralytics

Expand Down

0 comments on commit 96bc51a

Please sign in to comment.