11 lines
310 B
Bash
11 lines
310 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" ./
|
|
# If build with extensions fails, you can run this line to build without extensions
|
|
# pip install -v --no-cache-dir ./
|
|
cd $ROOT
|
|
rm -rf apex
|