GitHubJKin

Is it necessary to use Ampere GPU?

2

运行环境是colab，gpu是tsla T4，torch版本是2.0，CUDA版本是11.8，训练模型的时候出现这个报错：ValueError: Your setup doesn't support bf16/gpu. You need torch>=1.10, using Ampere GPU with cuda>=11.0