Skip to content

Commit b444a9b

Browse files
authored
Bump torch and torchvision to current versions (#25)
## Summary and motivation The torch version here is >3 years out of date. This is what's triggering the dependabot security alerts. Since we're still at the start of the project, let's bump to the latest stable versions, so we can 1) get the security fixes and 2) benefit from other improvements (e.g. memory efficiency, more integrated CUDA support) over the past few years. ## Testing I ran `uv run python main train ...` locally simply to confirm that it runs. Have not run it on a GPU yet. @hanaol would be good to test on your cluster to make sure the large upgrade jump doesn't break anything.
1 parent 3f3c677 commit b444a9b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

requirements.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
numpy<2
22
scikit-learn==1.7.2
3-
torch==1.11.0+cu113
4-
torchvision==0.12.0+cu113
3+
torch==2.9.0
4+
torchvision==0.24.0
55
pyscf==2.10.0
66
pyyaml==6.0.3
77
monty==2025.3.3

0 commit comments

Comments
 (0)