mirror of
https://github.com/norandom/log2ml.git
synced 2026-09-26 04:39:29 +00:00
auto install rapids for gpu acceleration on tpot-nn
This commit is contained in:
Vendored
+4
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user