Spaces:
Sleeping
Sleeping
| --index-url https://download.pytorch.org/whl/cu121 | |
| --extra-index-url https://pypi.org/simple | |
| ############################## | |
| # Core ML stack (CUDA build) # | |
| ############################## | |
| torch==2.5.0 | |
| torchvision==0.20.0 | |
| torchaudio==2.5.0 | |
| -f https://data.pyg.org/whl/torch-2.5.0+cu121.html | |
| torch_geometric | |
| pyg_lib | |
| torch_scatter | |
| torch_sparse | |
| torch_cluster | |
| torch_spline_conv | |
| ######################### | |
| # Numerics / science # | |
| ######################### | |
| numpy | |
| scipy | |
| pandas | |
| scikit-learn | |
| ######################### | |
| # Plotting & misc # | |
| ######################### | |
| matplotlib | |
| seaborn | |
| imageio | |
| meshio | |
| h5py | |
| tensorboard | |
| ######################### | |
| # Model / config helpers# | |
| ######################### | |
| timm | |
| einops | |
| omegaconf | |
| accelerate | |
| huggingface-hub | |
| lion-pytorch | |
| ######################### | |
| # 3D / VTK / Trame stack# | |
| ######################### | |
| vtk | |
| pyvista | |
| trame==3.12.0 | |
| trame-vtk==2.10.0 | |
| trame-vuetify==3.1.0 | |
| trame-server==3.6.2 | |
| trame-client==3.10.4 | |
| wslink | |
| ######################### | |
| # Web / misc # | |
| ######################### | |
| pathlib2>=2.3.0 | |
| typing-extensions>=4.0.0 | |
| transformers>=4.20.0 | |
| trimesh | |
| pytest | |
| requests | |
| PyYAML |