有人在50系显卡上训练过吗？

[cacard]

50系+最新pytorch nightly版本+cuda1.29 在跑高分辨率训练阶段时蓝屏。

[cacard]

Update:

python train_DINet_frame.py --augment_num=100 --mouth_region_size=128 --batch_size=80 --coarse2fine --coarse_model_path=./asserts/training_model_weight/frame_training_64/xxxxxx.pth --result_path=./asserts/training_model_weight/frame_training_128

Change the batch_size from 80 to 40 is OK.