挂载Google云端硬盘
1
2
3
4
5
6
7
8
|
from google.colab import drive
import os
drive.mount('/content/drive')
data_dir = 'drive/My Drive/dataset-resized'
output_dir = 'drive/My Drive/out_put'
logs = 'drive/My Drive/logs'
os.listdir(data_dir)
|
Drive already mounted at /content/drive; to attempt to forcibly remount, call drive.mount("/content/drive", force_remount=True).
查看被分配的GPU：
! nvidia-smi
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
|
Tue May 19 00:23:20 2020
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 440.82 Driver Version: 418.67 CUDA Version: 10.1 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 Tesla P4 Off | 00000000:00:04.0 Off | 0 |
| N/A 40C P8 7W / 75W | 0MiB / 7611MiB | 0% Default |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
| No running processes found |
+-----------------------------------------------------------------------------+
|