colab使用宝典
使用指南挂载谷歌云盘from google.colab import drivedrive.mount('/content/drive')安装依赖!pip install torch==1.7.0 torchvision!pip install opencv-python确认连到GPUimport tensorflow as tfdevice_name = tf.test.gpu_device_n
·
使用指南
挂载谷歌云盘
from google.colab import drive
drive.mount('/content/drive')
安装依赖
!pip install torch==1.7.0 torchvision
!pip install opencv-python
确认连到GPU
import tensorflow as tf
device_name = tf.test.gpu_device_name()
if device_name != '/device:GPU:0':
raise SystemError('没有发现GPU device')
print('Found GPU at: {}'.format(device_name))
# Found GPU at: /device:GPU:0
查看显卡型号
!/opt/bin/nvidia-smi
切换工作目录
import os
os.chdir("/content/drive/MyDrive/yolov5")
os.getcwd()
防止colab掉线(把这段代码放在控制台)
function ClickConnect(){
console.log("Working");
document
.querySelector("#top-toolbar > colab-connect-button")
.shadowRoot
.querySelector("#connect")
.click()
}
var id=setInterval(ClickConnect,5*60000) //5分钟点一次,改变频率把5换成其他数即可,单位分钟
//要提前停止,请输入运行以下代码: clearInterval(id)
最后感谢小伙伴们的学习噢~
更多推荐
已为社区贡献1条内容
所有评论(0)