Colab

colab pytorch安装： https://www.jianshu.com/p/000d2a9d36a0

以下为colab keras安装

!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}

!mkdir -p drive
!google-drive-ocamlfuse drive  -o nonempty

!pip install -q keras

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

pwd

cd drive/Colab/LSTM_FCN/LSTM_FCN

 

!python wm1.py

import tensorflow as tf
tf.test.gpu_device_name()

'/device:GPU:0'