Skip to content

Instantly share code, notes, and snippets.

@rohit-gupta

rohit-gupta/install_apex.sh

Created December 2, 2019 18:07
Show Gist options
  • Save rohit-gupta/ed9d73bd7346b0001680d1987427ec66 to your computer and use it in GitHub Desktop.
Save rohit-gupta/ed9d73bd7346b0001680d1987427ec66 to your computer and use it in GitHub Desktop.
Download ZIP
Install Nvidia Apex library
Raw
install_apex.sh
# Pascal Titan X, Titan Xp, 1080Ti, 1080, 1070Ti, 1070 ... = 6.1
# Volta Titan V = 7.0
# Turing Titan RTX, 2080Ti ... = 7.5
export TORCH_CUDA_ARCH_LIST="6.1;7.0;7.5" # Pascal, Volta and Turing on CRCV Cluster
git clone https://github.com/NVIDIA/apex
cd apex
pip install --upgrade --force-reinstall -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment