microsoft / computervision-recipes

Best Practices, code samples, and documentation for Computer Vision.
MIT License
9.47k stars 1.17k forks source link

Exception: It's not possible to apply those transforms to your dataset: expected scalar type Float but found Double #648

Open curryJ opened 3 years ago

curryJ commented 3 years ago

Description

Thanks for this great project. I run the 02_state_of_the_art.ipynb,but I get a Exception which is the title; My computer is cuda10.1 and GPU version is rtx2080ti.I update the version of pytorch and torchvision to 1.7.0 and 0.8.0 respectively

PatrickBue commented 3 years ago

Looks like a version conflict. You could try this conda environment, that should work: https://github.com/microsoft/computervision-recipes/blob/master/environment.yml