- 博客(4)
- 收藏
- 关注
原创 【解决30、40系列显卡使用mmrotate出现nvcc fatal: Unsupported gpu architecture ‘compute_86‘及size mismatch报错】
Facebook在2021年推出了pytorch1.8.0版本,而英伟达第一个型号的40系列显卡是在22年发布,因此需要更高的torch版本来支持更高版本的sm_86架构(即更高版本的cuda)。主要原因是mmrotate教程推荐的pytorch版本太低,因此需要保证当前使用的torch。和cuda版本适配30、40系列的显卡,而mmcv也必须重新安装适配torch的版本。
2024-08-25 11:33:23 252
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人