anh-khoa-nguyen commited on
Commit
c72df1c
·
1 Parent(s): df0f65d

Install CPU-only version of torch and torchvision

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -88,8 +88,8 @@ starlette==0.44.0
88
  sympy==1.13.3
89
  termcolor==2.4.0
90
  tifffile==2023.7.10
91
- torch==2.2.2
92
- torchvision==0.17.2
93
  tqdm==4.67.1
94
  typing_extensions==4.13.2
95
  urllib3==2.2.3
 
88
  sympy==1.13.3
89
  termcolor==2.4.0
90
  tifffile==2023.7.10
91
+ torch==2.2.2+cpu
92
+ torchvision==0.17.2+cpu
93
  tqdm==4.67.1
94
  typing_extensions==4.13.2
95
  urllib3==2.2.3