apex安装
1、环境:python3.7 torch1.13.1 cuda11.7
git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
2、环境:python3.7 cuda11.7
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch
本文提供了两种Apex安装方法,一种是针对python3.7、torch1.13.1和CUDA11.7环境,使用gitclone和pip安装,包含cpp_ext和cuda_ext选项。另一种方法是通过conda安装pytorch1.11.0、torchvision0.12.0和torchaudio0.11.0,以及cudatoolkit11.3。
6890

被折叠的 条评论
为什么被折叠?



