marius/log2ml
1
0
Fork 0
mirror of https://github.com/norandom/log2ml.git synced 2024-12-04 22:53:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

auto install rapids for gpu acceleration on tpot-nn

Browse Source
This commit is contained in:
Marius Ciepluch 2024-07-30 12:08:23 +02:00 committed by GitHub
parent da85a7f61f
commit e98fe3fd53
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
dependencies/install.sh vendored
View File

@ -6,6 +6,10 @@ if [ -d "/content" ]; then
echo "Installing dependencies"
pip install -r "https://raw.githubusercontent.com/norandom/log2ml/main/dependencies/requirements.gpu.txt"
# https://docs.rapids.ai/deployment/stable/platforms/colab/
git clone https://github.com/rapidsai/rapidsai-csp-utils.git
python rapidsai-csp-utils/colab/pip-install.py
else
echo "This does not appear to be a Google Colab environment."
fi