前提:要装了PYTORCH。
第一步:
下载apex:https://github.com/NVIDIA/apex
得到压缩包apex-master.zip,下载到哪里无所谓,自己解压就行了,解压到哪里没关系。
第二步:
打开requirements.txt
,看清要求:
第三步:
安装requirements.txt
里面要求的东西;
我用的是anaconda
,打开anaconda prompt
,进入自己的虚拟环境(自己随便)activate xxx
(自己环境名字),然后`
pip install cxxfilt
pip install tqdm
pip install numpy
pip install PyYAML
pip install pytest
安装好了就开始安装apex.
第四步:
安装apex;
在之前的环境下,进入到之前解压的apex-master目录下:
我的是在D盘,然后命令就是:
(pytorch) C:\Users\wanghao>D:
# D: 这是进入D盘,(pytorch)这是我的虚拟环境名字
(pytorch) D:\>cd D:\conda\apex-maste
# cd D:\conda\apex-maste,cd命令进入要进入的文件夹
(pytorch) D:\conda\apex-master>python setup.py install
# python setup.py install最后运行安装就行了。