大概率就是pytorch版本不对,可以用1.9.0版本的pytorch:
pip install torch1.9.0 torchvision0.10.0 torchaudio==0.9.0
RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling `cublasSg
最新推荐文章于 2024-04-13 13:18:04 发布
大概率就是pytorch版本不对,可以用1.9.0版本的pytorch:
pip install torch1.9.0 torchvision0.10.0 torchaudio==0.9.0