apex を Windows インストールする

インストール

git clone https://github.com/NVIDIA/apex
cd apex
pip install -r requirements.txt

pip install -v --no-cache-dir --no-build-isolation .

インストールしていない場合はあらかじめインストールしてください

PyTorch

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124

NVIDIA CUDA ツールキット のインストール pytorchとバージョンを合わせます

https://developer.nvidia.com/cuda-toolkit-archive

Python

Posted by eightban