search

ROCm / pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration
http://pytorch.org
Other
219 stars 51 forks source link

[release/2.0] Remove references to `pkg_resources.packaging` #1423

Closed jithunnair-amd closed 4 months ago

jithunnair-amd commented 4 months ago

Tested successfully via http://rocm-ci.amd.com/job/rocm-pytorch-manylinux-wheel-builder/295/

Tested that TorchVersion doesn't give an import error for packaging.

root@588a082ce4d1:/# pip list | grep setuptools
setuptools          70.0.0
root@588a082ce4d1:/# python -c 'from pkg_resources import packaging'
<string>:1: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
Traceback (most recent call last):
  File "<string>", line 1, in <module>
ImportError: cannot import name 'packaging' from 'pkg_resources' (/usr/local/lib/python3.9/dist-packages/pkg_resources/__init__.py)
root@588a082ce4d1:/# python -c 'import torch.torch_version; print(torch.torch_version.TorchVersion("1.0.0"))'
1.0.0
root@588a082ce4d1:/# python -c 'import torch.torch_version; print(torch.torch_version.TorchVersion(torch._vendor.packaging.version.Version("1.0.0")))'
1.0.0