InstantMesh icon indicating copy to clipboard operation
InstantMesh copied to clipboard

Fail to run the code on my local GPUs due to PyTorch and CUDA version.

Open snowflakewang opened this issue 1 year ago • 2 comments

Hello, thank you for your brilliant work on feed-forward-style Image-to-3D generation! I have tried your work on the HuggingFace demo and it works well! I want to run your code on my local GPUs. However, due to some inevitable problems, my GPUs do not support the newest 12.x CUDA version and cannot be updated. So do you have any methods to run the code based on PyTorch 1.1x (such as 1.12, 1.13) and CUDA 11.x (such as 11.4, 11.6)?

Thanks a lot! :)

snowflakewang avatar Apr 23 '24 05:04 snowflakewang

I'm sorry but we haven't tested on older pytorch or CUDA versions. It seems that using older pytorch version (<= 2.0) may lead to errors in the attention operation #8. However, pytorch 2.1+ is indeed compatible with CUDA 11.x, the pytorch offical site has provided the command for installing the newest pytorch with CUDA 11.x.

bluestyle97 avatar Apr 23 '24 07:04 bluestyle97

I'm sorry but we haven't tested on older pytorch or CUDA versions. It seems that using older pytorch version (<= 2.0) may lead to errors in the attention operation #8. However, pytorch 2.1+ is indeed compatible with CUDA 11.x, the pytorch offical site has provided the command for installing the newest pytorch with CUDA 11.x.

please do we need than 8go GPU to run the project ?

joshkiller avatar Apr 25 '24 00:04 joshkiller

@joshkiller We did not make a thorough test but I don't think 8GB is enough.

bluestyle97 avatar May 05 '24 09:05 bluestyle97