[Deep Learning][AI] 在 google colab 上 安裝 PyTorch

Bryan Yang
A multi hyphen life
2 min readMar 3, 2018

Google 開放好多神器幫助大家玩 AI

其實我不太喜歡把 Deep Learning 說成 AI 啦…不過大家都在用就跟風一下.

colab

https://colab.research.google.com/notebooks/welcome.ipynb

Google 放出來的神器之一,可以直接在 Google 免費提供的 container 上建立 ipython notebook,並且將檔案直接存在個人的 Google Drive 裡.對於學習 python 相當的方便.

另外一個優點就是支援 GPU GPU GPU!免費的 k80 讓大家用到爽.

規格

CPU: Intel(R) Xeon(R) CPU @ 2.20GHz 2 cores
MEM: 12G
HDD: 50G

Install PyTorch

PyTorch 懶得介紹了,Facebook 主打的 DL 框架,使用 Python.

因為安裝 PyTorch 需要考量 OS, python, 以及 Cuda 的版本,安裝的方法如下:

!pip3 install http://download.pytorch.org/whl/cu80/torch-0.3.0.post4-cp36-cp36m-linux_x86_64.whl!pip3 install torchvision

就這樣!

使用

這樣就ok了~

--

--

Bryan Yang
A multi hyphen life

Data Engineer, Data Producer Manager, Data Solution Architect