Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

Installation instructions for Colab environments

See original GitHub issue

Description

I’m wondering if you could installation instructions optimized for Colab. Colab takes seconds to spin up, but cupy installation is >10 minutes, so it’s a barrier to some quick experimentation.

I found the following works, but it takes >10 minutes, mostly due to cudatoolkot download. Cuda should already be available on Colab environments…

!python -m pip install -U setuptools pip
!python -m cupyx.tools.install_library --cuda 11.2 --library cutensor

!pip install -q condacolab
import condacolab
condacolab.install()

!conda --version
 !conda install -c conda-forge cupy cudnn cutensor nccl

Screen Shot 2022-01-10 at 4 04 02 PM

Additional Information

No response

Issue Analytics

State:
Created 2 years ago
Comments:6 (3 by maintainers)

Top GitHub Comments

2reactions

yaroslavvbcommented, Jan 11, 2022

You are right…silly mistake on my part, it does have Cupy after all! Thanks for the upgrade command

1reaction

emcastillocommented, Jan 11, 2022

colab has cupy-cuda111==9.4.0

You can do !pip install cupy-cuda111 --upgrade and it should work, I just verified it.

Top Results From Across the Web

Google CoLab Tutorial — How to setup a Pytorch ... - Medium

1. Create a Colab document · 2. Edit and run a Colab document · 3. How to Install Python Library in Colab ·...

Installation in the Google Colaboratory environment

Step 1: Create new Notebook in the Google Research Colaboratory¶ · Step 2: Create a new Code Cell, with the following code¶ ·...

Appendix A. Setting up environments on Google Colaboratory

This appendix provides instructions for setting up an environment to run the code ... Using pip to install the AutoKeras package in a...

Python Installation - Colab Notebook | Google Earth Engine

This guide demonstrates setup and testing with a new Colab notebook, ... Follow the instructions printed to the cell to complete this step....

Local runtimes - Google Colab

Setup instructions · Step 1: Install Jupyter · Step 2: Install and enable the jupyter_http_over_ws jupyter extension (one-time) · Step 3: Start server...