Skip to content

Commit 6406778

Browse files
committed
update pytorch version
1 parent 1923d40 commit 6406778

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

environment.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,5 +7,5 @@ dependencies:
77
- pandas
88
- scikit-learn
99
- matplotlib
10-
- pytorch
10+
- pytorch>=1.7.0
1111
- cpuonly

requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,4 +5,4 @@ pandas
55
scikit-learn
66
matplotlib
77
-f https://download.pytorch.org/whl/torch_stable.html
8-
torch==1.3.1+cpu
8+
torch>=1.7.0+cpu

0 commit comments

Comments
 (0)