Colab每次跑程序必用代码记录

Posted 2021-02-19 lxwen

1.记得选择使用GPU

# 授权绑定Google Drive
!apt-get install -y -qq software-properties-common python-software-properties module-init-tools
!add-apt-repository -y ppa:alessandro-strada/ppa 2>&1 > /dev/null
!apt-get update -qq 2>&1 > /dev/null
!apt-get -y install -qq google-drive-ocamlfuse fuse
from google.colab import auth
auth.authenticate_user()
from oauth2client.client import GoogleCredentials
creds = GoogleCredentials.get_application_default()
import getpass
!google-drive-ocamlfuse -headless -id={creds.client_id} -secret={creds.client_secret} < /dev/null 2>&1 | grep URL
vcode = getpass.getpass()
!echo {vcode} | google-drive-ocamlfuse -headless -id={creds.client_id} -secret={creds.client_secret}

2.安装必要的库（还不确定必须需要）

# 安装最新版本Keras

# https://keras.io/

!pip install keras

# 指定版本安装

!pip install keras==2.1.5

# 安装 OpenCV

# https://opencv.org/

!apt-get -qq install -y libsm6 libxext6 && pip install -q -U opencv-python

# 安装 Pytorch

# http://pytorch.org/

!pip install -q http://download.pytorch.org/whl/cu75/torch-0.2.0.post3-cp27-cp27mu-manylinux1_x86_64.whl torchvision

# 安装 XGBoost

# https://github.com/dmlc/xgboost

!pip install -q xgboost

# 安装 7Zip

!apt-get -qq install -y libarchive-dev && pip install -q -U libarchive

# 安装 GraphViz 和 PyDot

!apt-get -qq install -y graphviz && pip install -q pydot

!pip install jieba

!pip install h5py

 

3.进入文件并查看

# 挂载Google driver，指定Google Drive云端硬盘的根目录，名为drive

!mkdir -p drive

!google-drive-ocamlfuse drive

# 指定当前的工作目录

# 此处为google drive中的文件路径,drive为之前指定的工作根目录，要加上

import os

os.chdir("drive/BreastCancer_FinalVersion")

!ls

 

4. 运行

!python trainAlexnet.py

 

使用TensorFlow的实现，tf.keras。这个适用于TF 2。
例如你需要使用tf.keras，必须确保使用正确的导入：
from tensorflow import keras
而不是直接使用：import keras

同样，在要使用keras下的其他模块时：
from tensorflow.keras import layers
而不是使用 from keras import layers

https://www.jianshu.com/p/c09ef6597b8b