apex安装报错

大部分apex安装报错主要是因为环境不适配,例如cuda版本不适配torch,安装前注意查看需求的cuda版本:

# pytorch内的cuda版本
import torch
torch.version.cuda

具体安装参考:Nvidia Apex安装与简单使用_豆芽菜-CSDN博客

猜你喜欢

转载自blog.csdn.net/dou3516/article/details/121284311