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)