colab 连接 google drive
前两句 mount，后两句切换路径并查看当前路径，pip
等命令执行前加!
1
2
3
4
| from google.colab import drive
drive.mount('/content/drive')
%cd /content/drive/MyDrive/"Colab Notebooks"/"blog-dataset-vs-accuracy"
%pwd
|
查看可用 GPU/CPU 容量
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
| !ln -sf /opt/bin/nvidia-smi /usr/bin/nvidia-smi
!pip install gputil
!pip install psutil
!pip install humanize
import psutil
import humanize
import os
import GPUtil as GPU
GPUs = GPU.getGPUs()
gpu = GPUs[0]
def printm():
process = psutil.Process(os.getpid())
print("Gen RAM Free: " + humanize.naturalsize( psutil.virtual_memory().available ), " | Proc size: " + humanize.naturalsize( process.memory_info().rss))
print("GPU RAM Free: {0:.0f}MB | Used: {1:.0f}MB | Util {2:3.0f}% | Total {3:.0f}MB".format(gpu.memoryFree, gpu.memoryUsed, gpu.memoryUtil*100, gpu.memoryTotal))
printm()
|
查看 GPU 型号：!/opt/bin/nvidia-smi
