Colab+pytoch使用

一、colabに関する問題
1、colab.reearch.google.com 
新しいノートを開きます.「修正」－>「ノート設定」では、「ハードウェア加速器」を「GPU」に設定します.

from tensorflow.python.client import device_lib
device_lib.list_local_devices()

または

import torch
torch.cuda.get_device_name(0)

使用されているのは「Tesla T 4」であることが分かります.
2、Google driveのデータを使う
まず、ローカルデータセットをGoogle driveにアップロードし、「/content/gdrive/My Drive/data」に保存します.

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

コードの中のfile_をnameはGoogle driveの保存先を指します.
3、外部ファイルを読み書きする
参照https://blog.csdn.net/heliucs/article/details/84644372
https://colab.research.google.com/notebooks/io.ipynb#scrollTo=BaCkyg5CV5jF
(1)ローカルファイルをアップロードすると、インタフェースが現れて、ファイルをアップロードすることを選択します.

from google.colab import files
uploaded = files.upload()
for fn in uploaded.keys():
    print('User uploaded file "{name}" with length {length} bytes'.format(name=fn, length=len(uploaded[fn])))

(2)colabはファイルをローカルにダウンロードする(「ダウンロード」のようなデフォルトのダウンロード場所にダウンロードする)

from google.colab import files
with open('example.txt', 'w') as f:
    f.write('some content')
files.download('example.txt')

二、pytouch関連問題
1、ローカルにpytouchをインストールする

conda install --use-local F:/pytorch-0.4.0-py35_cuda8.0.61_cudnn7.1.2_1.tar.bz2

2、「cuDNN error:CUDNNNN＿STATUSchu＿INTERNALUERROR」が現れます.
ソリューション:グラフィックスカードを指定します.https://blog.csdn.net/u014176855/article/details/80994991

import os
os.environ["CUDA_VISIBLE_DEVICES"]="0"

(続き)