参见windows 安装apex_Nvidia Apex安装
重点:
1、在安装前先检查一下,电脑的cuda版本和pytorch内的cuda版本是否一样,不一样的话就把低版本的进行升级;
$ git clone https://github.com/NVIDIA/apex
$ cd apex
2、在保证cuda版本一致的前提下进入apex文件夹,使用命令:python setup.py install 即可安装成功
参见windows 安装apex_Nvidia Apex安装
重点:
1、在安装前先检查一下,电脑的cuda版本和pytorch内的cuda版本是否一样,不一样的话就把低版本的进行升级;
$ git clone https://github.com/NVIDIA/apex
$ cd apex
2、在保证cuda版本一致的前提下进入apex文件夹,使用命令:python setup.py install 即可安装成功