安装NVIDIA apex

1、下载源码

git clone https://github.com/NVIDIA/apex

2、cd apex

进入apex文件夹

3、安装相关的包(3条指令中,选其中一种进行安装包即可)

1 python setup.py install [--cuda_ext] [--cpp_ext]
2 pip install --upgrade setuptools
3 python setup.py install

详细解释

你可能感兴趣的:(python)