项目文件夹

文件
Rhett Ying 46a3fc2b5f [CI] updgrade pytorch version for benchmark CI (#5102)
* [CI] updgrade pytorch version for benchmark CI

* update build arguments

* updage

* updage

* upgrade torch to 1.13

* updage docker image

* update cmake args

* try with cu116_torch112

* update build

* update

* update

* update

* update docker image

* update

* update

* update

* update

* final update

* fix continue running

* update

* update

* update
2023-01-06 14:35:36 +08:00

20 行
238 B
Plaintext

--find-links https://download.pytorch.org/whl/torch_stable.html
torch==1.13.1+cu116
torchvision==0.14.1+cu116
torchmetrics
pytest
nose
numpy
cython
scipy
networkx
matplotlib
nltk
requests[security]
tqdm
awscli
torchtext
pandas
rdflib
ogb