Fix a few Pip dependency version mismatches (#435)

* Fix a few Pip dependency version mismatches

Signed-off-by: Abolfazl Shahbazi <[email protected]>

* Update oneCCL for Python 3.8 as well

Signed-off-by: Abolfazl Shahbazi <[email protected]>

---------

Signed-off-by: Abolfazl Shahbazi <[email protected]>

Author: ashahba

docker/Dockerfile

@@ -114,7 +114,7 @@ ARG PYTHON=python3
 ARG HOROVOD_WITH_PYTORCH=1
 ARG HOROVOD_WITHOUT_MXNET=1
 ARG HOROVOD_WITH_TENSORFLOW=1
-ARG HOROVOD_VERSION=0.28.0
+ARG HOROVOD_VERSION=0.28.1
 
 RUN apt-get update && apt-get install -y --no-install-recommends --fix-missing \
     build-essential \
@@ -128,7 +128,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends --fix-missing \
 
 RUN python -m pip install --no-cache-dir horovod==${HOROVOD_VERSION}
 
-ARG ONECCL_VERSION=1.13.0
+ARG ONECCL_VERSION=2.0.0
 ARG ONECCL_URL=https://developer.intel.com/ipex-whl-stable-cpu
 
 RUN python -m pip install --no-cache-dir oneccl_bind_pt==${ONECCL_VERSION} -f ${ONECCL_URL}

docker/docker-compose.yml

@@ -31,7 +31,7 @@ services:
       service: tlt-prod
     build:
       args:
-        HOROVOD_VERSION: 0.28.0
+        HOROVOD_VERSION: 0.28.1
         ONECCL_VERSION: 2.0.0
         ONECCL_URL: https://developer.intel.com/ipex-whl-stable-cpu
       target: tlt-dist-devel

setup.py

@@ -61,7 +61,7 @@ def get_framework_requirements(framework_name):
         EXTRA_PACKAGES['pytorch'] + \
         EXTRA_PACKAGES['tensorflow'],
       extras_require=EXTRA_PACKAGES,
-      python_requires=">3.8",
+      python_requires=">=3.8",
       entry_points={
         "console_scripts": [
             "tlt = tlt.tools.cli.main:cli_group"

tlt/distributed/pytorch/requirements.txt

@@ -1,4 +1,4 @@
 horovod[mpi,pytorch]==0.28.1
-oneccl-bind-pt @ https://intel-extension-for-pytorch.s3.amazonaws.com/torch_ccl/cpu/oneccl_bind_pt-1.13.0%2Bcpu-cp310-cp310-linux_x86_64.whl; python_version=='3.10'
-oneccl-bind-pt @ https://intel-extension-for-pytorch.s3.amazonaws.com/torch_ccl/cpu/oneccl_bind_pt-1.13.0%2Bcpu-cp38-cp38-linux_x86_64.whl; python_version=='3.8'
-oneccl-bind-pt @ https://intel-extension-for-pytorch.s3.amazonaws.com/torch_ccl/cpu/oneccl_bind_pt-1.13.0%2Bcpu-cp39-cp39-linux_x86_64.whl; python_version=='3.9'
+oneccl-bind-pt @ https://intel-extension-for-pytorch.s3.amazonaws.com/torch_ccl/cpu/oneccl_bind_pt-2.0.0%2Bcpu-cp38-cp38-linux_x86_64.whl#sha256=ef7bf156d6d29b7cc64c807a7c7f2ce0ad793c4da4acd65c2db959ea63845b0e; python_version=='3.8'
+oneccl-bind-pt @ https://intel-extension-for-pytorch.s3.amazonaws.com/torch_ccl/cpu/oneccl_bind_pt-2.0.0%2Bcpu-cp310-cp310-linux_x86_64.whl#sha256=982641f099e39c0ea5b8ea03860bb2d852e91b953cf7deba3719ff69e4a9857b; python_version=='3.10'
+oneccl-bind-pt @ https://intel-extension-for-pytorch.s3.amazonaws.com/torch_ccl/cpu/oneccl_bind_pt-2.0.0%2Bcpu-cp39-cp39-linux_x86_64.whl#sha256=165c2d5525cf18390f8636fb18ee174b74ad7e785de5a000a9c63008e50f2182; python_version=='3.9'