a | b/bin/install_apex | ||
---|---|---|---|
1 | #!/bin/bash |
||
2 | |||
3 | export TMPDIR=$HOME/tmp |
||
4 | mkdir -p $TMPDIR |
||
5 | rm -rf apex |
||
6 | git clone https://github.com/NVIDIA/apex |
||
7 | cd apex |
||
8 | pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./ || pip install -v --no-cache-dir ./ |
||
9 | cd .. |
||
10 | rm -rf apex |