ModuleNotFoundError: No module named 'dataclasses'

[FeliMe]

If you do not know the root cause of the problem / bug, and wish someone to help you, please
post according to this template:

🐛 Bugs / Unexpected behaviors

error when trying to import pytorch3d.renderer

Instructions To Reproduce the Issue:

I installed pytorch3d according to the INSTALL.md file (installed pytroch3d from Anaconda Cloud), when I try to import pytorch3d.renderer, I get the following error:

ModuleNotFoundError: No module named 'dataclasses'

Complete logs:

$ python
Python 3.6.10 |Anaconda, Inc.| (default, Jan  7 2020, 21:14:29)
[GCC 7.3.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.__version__
'1.4.0'
>>> torch.version.cuda
'10.0'
>>> import pytorch3d.renderer
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/meissen/miniconda3/envs/pytorch3d/lib/python3.6/site-packages/pytorch3d/renderer/__init__.py", line 19, in <module>
    from .mesh import (
  File "/home/meissen/miniconda3/envs/pytorch3d/lib/python3.6/site-packages/pytorch3d/renderer/mesh/__init__.py", line 4, in <module>
    from .rasterizer import MeshRasterizer, RasterizationSettings
  File "/home/meissen/miniconda3/envs/pytorch3d/lib/python3.6/site-packages/pytorch3d/renderer/mesh/rasterizer.py", line 5, in <module>
    from dataclasses import dataclass
ModuleNotFoundError: No module named 'dataclasses'
>>>

[nikhilaravi]

Hi @FeliMe dataclasses is available in Python 3.7! We are updating the anaconda cloud package to support Python 3.6 as well. For now can you try changing the python version to 3.7 and running the code again?

[FeliMe]

Updating to Python 3.7 worked, thanks @nikhilaravi !

[csverma610]

Thanks. I will do it.
Also for others, it will be nice to update the installation instructions.

[nikhilaravi]

@csverma610 this has been changed on master. We will be updating the conda package soon to incorporate this change as well.

[csverma610]

Hello, Should we use CudaTookit=10.0 or 10.1 or 10.2 ? Thanks

…

On Wed, Feb 19, 2020 at 1:12 PM Nikhila Ravi ***@***.***> wrote: @csverma610 <https://github.com/csverma610> this has been changed on master. We will be updating the conda package soon to incorporate this change as well. — You are receiving this because you were mentioned. Reply to this email directly, view it on GitHub <#31?email_source=notifications&email_token=AA2TTM2Y7TLQY22ON5PGXBTRDWODZA5CNFSM4KRVPYQ2YY3PNVWWK3TUL52HS4DFVREXG43VMVBW63LNMVXHJKTDN5WW2ZLOORPWSZGOEMJWD7I#issuecomment-588472829>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/AA2TTMYE3L2FB3DI6VZUDCTRDWODZANCNFSM4KRVPYQQ> .

[bottler]

We don't yet have builds for 10.2, because pytorch does not have builds for 10.2 yet. I recommend 10.1 as it's the latest we do have builds for.

[vivekslair]

Incase you still need to use python 3.6 you might want to manually install dataclasses. In Mac pip usually fails and hence do a conda install dataclasses.