Skip to content

Instantly share code, notes, and snippets.

@BlurryLight
Last active December 27, 2019 05:13
Show Gist options
  • Save BlurryLight/2687663b243e2514ce70aeaa527c04ce to your computer and use it in GitHub Desktop.
Save BlurryLight/2687663b243e2514ce70aeaa527c04ce to your computer and use it in GitHub Desktop.
colab/ kaggle snippets

colab mount drive

from google.colab import drive
drive.mount('/content/gdrive')

install pytorch 1.2

!pip install torch==1.2.0 torchvision==0.4.0 -f https://download.pytorch.org/whl/torch_stable.html

install pytorch apex

%%writefile setup.sh

git clone https://github.com/NVIDIA/apex
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./apex

!sh setup.sh
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment