fvcore
lightning
netCDF4
numpy
pandas
psutil
pydot
torch
torchvision
tqdm
typing_extensions
rocrate
prov
zarr<3
lxml
rdflib<7

[amd]
pyamdgpuinfo==2.1.*
amdsmi

[apple]
apple_gpu==0.3.*

[nvidia]
nvitop==1.3.*
nvidia-ml-py
