兩種方法:
1、去github下下載apex ,之后安裝到你的python環境下,我的安裝路徑:E:\Anaconda\anaconda\envs\pytorch\Lib\site-packages
注:建議用git下載
1、git clone git@github.com:NVIDIA/apex.git
2、開cmd命令窗口,切換到apex所在的文件夾
3、python setup.py install
如果出現:ModuleNotFoundError錯誤,建議使用以下代碼:
pip install --upgrade setuptools
2、需要你有anaconda環境,因為需要用到conda命令,我是通過這個方法安裝好的。
conda install -c conda-forge nvidia-apex