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 !pip install keras # 安装 OpenCV !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
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