update-runtime.sh 941 B

123456789101112131415161718192021
  1. #!/bin/bash
  2. wget -qO- https://micromamba.snakepit.net/api/micromamba/linux-64/latest | tar -xvj bin/micromamba
  3. if [ ! -f "conda/envs/linux/bin/python" ]; then
  4. bin/micromamba create --no-shortcuts -r conda -n linux -f environment.yaml -y
  5. fi
  6. bin/micromamba create --no-shortcuts -r conda -n linux -f environment.yaml -y
  7. bin/micromamba install -r conda -n linux gxx=10 -c conda-forge -y
  8. bin/micromamba run -r conda -n linux pip install -r requirements.txt
  9. # Make it so the correct NVCC is found. Will do a quick search that shouldn't take more than a few seconds and get the first result.
  10. export CUDA_HOME=$(find / -type d -path "*/conda/envs/linux" 2>/dev/null | head -n 1)
  11. read -p "Do you want to install aphrodite from source? (y/n): " answer
  12. case ${answer:0:1} in
  13. y|Y )
  14. bin/micromamba run -r conda -n linux pip install -e .
  15. ;;
  16. * )
  17. bin/micromamba run -r conda -n linux pip install aphrodite-engine
  18. ;;
  19. esac