junyanz / VON

[NeurIPS 2018] Visual Object Networks: Image Generation with Disentangled 3D Representation.
http://von.csail.mit.edu
Other
530 stars 90 forks source link

Does it support pytorch 1.0.0 or later verison? #18

Open zengqixun123 opened 5 years ago

zengqixun123 commented 5 years ago

Thanks a lot and waiting for your answer!

junyanz commented 5 years ago

It should support PyTorch 1.0. But let us know if you find any issues.

scamper07 commented 4 years ago

Hi, I'm unable to get it running with PyTorch 1.3 I get the following error

dataset [ImageAndDFDataset] was created enable cudnn benchmark initialization method [kaiming] loading model from /home/student/Projects/VON/scripts/../final_models/models_3D/car_df_G_3D.pth initialization method [kaiming] loading model from /home/student/Projects/VON/scripts/../final_models/models_2D/car_df/latest_net_G_AB.pth initialization method [kaiming] loading model from /home/student/Projects/VON/scripts/../final_models/models_2D/car_df/latest_net_E.pth Traceback (most recent call last): File "/home/student/Projects/VON/render_module/render_sketch.py", line 2, in <module> from .vtn.vtn.functions import grid_sample3d, affine_grid3d File "/home/student/Projects/VON/render_module/vtn/vtn/functions/__init__.py", line 2, in <module> from .grid_sample3d import grid_sample3d File "/home/student/Projects/VON/render_module/vtn/vtn/functions/grid_sample3d.py", line 3, in <module> from .._ext import vtn_lib File "/home/student/Projects/VON/render_module/vtn/vtn/_ext/vtn_lib/__init__.py", line 2, in <module> from torch.utils.ffi import _wrap_function File "/home/student/anaconda3/envs/von/lib/python3.7/site-packages/torch/utils/ffi/__init__.py", line 1, in <module> raise ImportError("torch.utils.ffi is deprecated. Please use cpp extensions instead.") ImportError: torch.utils.ffi is deprecated. Please use cpp extensions instead.

Any way to fix this? I cannot use version 0.4.1 since i have the new Quadro RTX 4000 card and it needs CUDA10 and all newer version packages

ztzhang commented 4 years ago

Hi, pytorch has changed its ffi interface and sorry we currently do not have the bandwidth to do a full upgrade. I'll see if I can find time to do this during the holidays.

scamper07 commented 4 years ago

Thank you for your reply. I got it working with PyTorch 0.4.1 CUDA 9.2