NVIDIA apex安装

如果是在docker中安装，步骤如下：

（1）docker pull pytorch/pytorch:1.8.1-cuda10.2-cudnn7-devel。这里必须是-devel后缀！因为编译时需要NVCC，只有-devel中有NVCC。cuda11.2有问题。

（2）按着NVIDIA/apex中的说明安装即可：

git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

 

posted @ 2022-02-22 11:37  kuaqi  阅读(1054)  评论(0)  编辑  收藏  举报