Colab 使用技巧

GPU 查看和使用

# 查看 GPU 显存信息 !/opt/bin/nvidia-smi # 查看 gpu 及使用 gpu import tensorflow as tf print(tf.test.gpu_device_name()) tf.device('/gpu:0')

CUDA 版本

安装 cuda 9

cuda 和 tf 版本对应

!cat /usr/local/cuda/version.txt !wget https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64-deb !dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64-deb !apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub !apt-get update !apt-get install cuda=9.0.176-1

Colab 断联解决方法

白嫖真是爽 ——google colab 总是断连解决方法

F12 打开 notebook 界面，进入 Console 选项卡，输入 JS 命令

function ClickConnect(){ console.log("Working"); document.querySelector("colab-toolbar-button#connect").click() } setInterval(ClickConnect,60000)