Problem for running the Trained Model

NVlabs / latentfusion

LatentFusion: End-to-End Differentiable Reconstruction and Rendering for Unseen Object Pose Estimation

https://arxiv.org/pdf/1912.00416.pdf

Other

213 stars 34 forks source link

Open SHZ2006 opened 3 years ago

SHZ2006 commented 3 years ago

Hi, when I want to run this line of code from latentfusion.recon.inference import LatentFusionModel

I am getting this error: cannot import name 'autocast' from 'torch.cuda.amp'

This installation also doesn't help conda install pytorch torchvision cudatoolkit=10.2 -c pytorch-nightly

It seems the nightly version is not compatible with other toolboxes that you installed in the environment by environment.yml

keunhong commented 3 years ago

Hi,

I don't think you need the nightly version anymore since PyTorch stable has it included.

SHZ2006 commented 3 years ago

But still I am getting this error cannot import name 'autocast' from 'torch.cuda.amp'