Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
recommended_gpu.txt 289 Bytes
Newer Older
Yuge Zhang's avatar
Fix pipeline dependency (#3926)  
Yuge Zhang committed
1
2
3
4 
# Recommended because some non-commonly-used modules/examples depend on those packages.

-f https://download.pytorch.org/whl/torch_stable.html
tensorflow
Yuge Zhang's avatar
Use python 3.9 on pipeline (#3881)  
Yuge Zhang committed
5 
keras == 2.4.3
Yuge Zhang's avatar
Upgrade pytorch dependency (#4344)  
Yuge Zhang committed
6
7 
torch == 1.10.0+cu111
torchvision == 0.11.1+cu111
Yuge Zhang's avatar
Remove NAS UI (#4088)  
Yuge Zhang committed
8 
pytorch-lightning >= 1.4.2
Yuge Zhang's avatar
Fix pipeline dependency (#3926)  
Yuge Zhang committed
9
10
11
12
13 
onnx
peewee
graphviz
gym
tianshou >= 0.4.1