search

guoqincode / Open-AnimateAnyone

Unofficial Implementation of Animate Anyone
2.85k stars 230 forks source link

Problem with training with H100 #114

Closed Valentino-unova closed 4 months ago

Valentino-unova commented 5 months ago

Hi! I got this incompatibility problem with PyTorch when training with H100:

NVIDIA H100 PCIe with CUDA capability sm_90 is not compatible with the current PyTorch installation. The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70 sm_75 sm_80 sm_86.

Screenshot 2024-02-12 at 18 03 43

The training is not started properly. Do you have any idea which PyTorch version in this case would fit with H100 while making the training work? Thanks a lot!

guoqincode commented 5 months ago

If you are using an H100, I recommend Torch 2.x