diff --git a/.bazelversion b/.bazelversion index 712bd5a680..8cf6caf561 100644 --- a/.bazelversion +++ b/.bazelversion @@ -1 +1 @@ -3.3.1 \ No newline at end of file +3.4.1 \ No newline at end of file diff --git a/WORKSPACE b/WORKSPACE index 671f0c3e33..dbcbc3cf45 100644 --- a/WORKSPACE +++ b/WORKSPACE @@ -53,16 +53,16 @@ http_archive( name = "libtorch", build_file = "@//third_party/libtorch:BUILD", strip_prefix = "libtorch", - urls = ["https://download.pytorch.org/libtorch/cu102/libtorch-cxx11-abi-shared-with-deps-1.5.1.zip"], - sha256 = "cf0691493d05062fe3239cf76773bae4c5124f4b039050dbdd291c652af3ab2a" + urls = ["https://download.pytorch.org/libtorch/cu102/libtorch-cxx11-abi-shared-with-deps-1.6.0.zip"], + sha256 = "fded948bd2dbee625cee33ebbd4843a69496729389e0200a90fbb667cdaeeb69" ) http_archive( name = "libtorch_pre_cxx11_abi", build_file = "@//third_party/libtorch:BUILD", strip_prefix = "libtorch", - sha256 = "818977576572eadaf62c80434a25afe44dbaa32ebda3a0919e389dcbe74f8656", - urls = ["https://download.pytorch.org/libtorch/cu102/libtorch-shared-with-deps-1.5.1.zip"], + sha256 = "141bb229f4bbf905541096cf8705785e7b0c79e37ca1e5db9d372730b1b9abd7", + urls = ["https://download.pytorch.org/libtorch/cu102/libtorch-shared-with-deps-1.6.0.zip"], ) # Download these tarballs manually from the NVIDIA website diff --git a/py/build_whl.sh b/py/build_whl.sh index 87a58feb05..aca11f4885 100755 --- a/py/build_whl.sh +++ b/py/build_whl.sh @@ -6,12 +6,6 @@ cd /workspace/TRTorch/py export CXX=g++ -build_py35() { - /opt/python/cp35-cp35m/bin/python -m pip install -r requirements.txt - /opt/python/cp35-cp35m/bin/python setup.py bdist_wheel - #auditwheel repair --plat manylinux2014_x86_64 -} - build_py36() { /opt/python/cp36-cp36m/bin/python -m pip install -r requirements.txt /opt/python/cp36-cp36m/bin/python setup.py bdist_wheel @@ -30,7 +24,6 @@ build_py38() { #auditwheel repair --plat manylinux2014_x86_64 } -build_py35 build_py36 build_py37 build_py38 \ No newline at end of file diff --git a/py/requirements.txt b/py/requirements.txt index 33c8f475c9..46e982e6b5 100644 --- a/py/requirements.txt +++ b/py/requirements.txt @@ -1 +1 @@ -torch==1.5.1 \ No newline at end of file +torch==1.6.0 \ No newline at end of file diff --git a/py/setup.py b/py/setup.py index d40a5cfc96..86c0dd5726 100644 --- a/py/setup.py +++ b/py/setup.py @@ -196,7 +196,7 @@ def run(self): long_description=long_description, ext_modules=ext_modules, install_requires=[ - 'torch==1.5.1', + 'torch==1.6.0', ], setup_requires=[], cmdclass={ @@ -210,7 +210,7 @@ def run(self): license="BSD", packages=find_packages(), classifiers=[ - "Development Status :: 3 - Alpha", + "Development Status :: 4 - Beta", "Environment :: GPU :: NVIDIA CUDA", "License :: OSI Approved :: BSD License", "Intended Audience :: Developers", @@ -224,7 +224,7 @@ def run(self): "Topic :: Software Development", "Topic :: Software Development :: Libraries" ], - python_requires='>=3.5', + python_requires='>=3.6', include_package_data=True, package_data={ 'trtorch': ['lib/*.so'], diff --git a/tests/py/requirements.txt b/tests/py/requirements.txt index 26cd815aec..ac6580af90 100644 --- a/tests/py/requirements.txt +++ b/tests/py/requirements.txt @@ -1 +1 @@ -torchvision==0.6.0 \ No newline at end of file +torchvision==0.7.0 \ No newline at end of file