谷歌colab使用笔记

1.挂载谷歌云盘

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

2.查看及更改当前工作目录

import os
print(os.getcwd())
os.chdir("/content/drive/My Drive/pointnet-master/sem_seg/")
print(os.getcwd())

3.检测是否用了GPU以及看GPU环境的常用命令

import tensorflow as tf
tf.test.gpu_device_name()

如果结果为空，则不能使用GPU，如果结果为/device:GPU:0，则使用!/opt/bin/nvidia-smi查看显存情况

4.查看显卡内存使用上限

from tensorflow.python.client import device_lib
device_lib.list_local_devices()

Tip: ! python train.py 这样的命令可以当Ubantu控制台用