facebookresearch / vissl

VISSL is FAIR's library of extensible, modular and scalable components for SOTA Self-Supervised Learning with images.
https://vissl.ai
MIT License
3.24k stars 330 forks source link

Official Support for Pytorch 1.10 #529

Open iseessel opened 2 years ago

iseessel commented 2 years ago

🚀 Feature

Officially support Pytorch 1.10.

Notes

User wants to try to use: https://developer.nvidia.com/blog/accelerating-inference-up-to-6x-faster-in-pytorch-with-torch-tensorrt/

Building and releasing APEX will be required for full VISSL support; however, most functionality should be available without APEX.

Sync with @iseessel first and then @bottler for any more information regarding this.

iseessel commented 2 years ago

FYI @pietroastolfi

iseessel commented 2 years ago

use the code torch.version[0:5:2] to remove the dots from the version string and get e.g. 191 from e.g. PyTorch 1.9.1. But This fails if you need to get 1100 from PyTorch 1.10.0.

When/if you make apex builds for PyTorch 1.10, you'll need to update Install.md, dev/packaging/apex_pip/README.md, docs/source/installation.rst and some tutorials

Also S3 details have changed since fs3cmd has been deprecated.

Sync with @iseessel first and then @bottler for any more information regarding this.