lightning/tests/install_AMP.sh

10 lines
224 B
Bash

#!/usr/bin/env bash
ROOT=$PWD
git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
pip install -v --no-cache-dir ./
cd $ROOT
rm -rf apex