樓主前一階段在做視頻插幀算法應(yīng)用,鑒于在自己的本子上跑代碼是在太慢,又不好意思在跑路后還是用學(xué)院的服務(wù)器賬號,所以翻來覆去學(xué)會了在谷歌使用免費的算力進(jìn)行模型訓(xùn)練和使用。在開始使用前,請您準(zhǔn)備自己的Google賬號并熟悉Jupyter Notebook的使用
1. Colab簡介
什么是 Colab?
借助 Colaboratory(簡稱 Colab),您可在瀏覽器中編寫和執(zhí)行 Python 代碼:
- 無需任何配置
- 免費使用 GPU和TPU
- 輕松共享
無論您是一名學(xué)生、數(shù)據(jù)科學(xué)家還是 AI 研究員,Colab 都能夠幫助您更輕松地完成工作。
PS:Colab免費版一次只享有12個小時的使用時間,并且使用期間不能關(guān)閉瀏覽器。如果是富哥,可以開通付費訂閱,擁有更高級的處理器,更大的內(nèi)存和更長的運行時間,并且在瀏覽器關(guān)閉后仍能繼續(xù)運行代碼。
2. 使用準(zhǔn)備
2.1 科學(xué)上網(wǎng)
Colab是谷歌旗下產(chǎn)品,由于眾所周知的原因,谷歌的服務(wù)在中國大陸地區(qū)不能直接訪問,所以請各顯神通,樓主就不細(xì)說了。
2.2 Google Drive
由于Colab的服務(wù)器是共享機制,所以上面的數(shù)據(jù)和文件在離開Colab十二個小時會自動刪除,為了保存我們的代碼和數(shù)據(jù),我們需要在Colab上掛載谷歌云盤(就是中間那個三角形辣)
在Google云盤下,新建colab文件夾以放置我們要在colab上運行的文件。并將要運行的工程文件上傳到該文件夾下。
3. Colab 使用
3.1 云硬盤掛載
找到上傳的項目文件及,點擊右鍵打開方式,選擇關(guān)聯(lián)更多應(yīng)用。
搜索Colaboratory,使用Colab打開。
打開后會自動生成一個.ipynb文件,我們在這個文件里進(jìn)行命令輸入和執(zhí)行
使用如下命令掛載谷歌云盤,掛載到/content/drive/目錄下
from google.colab import drive
drive.mount('/content/drive')
使用如下命令將當(dāng)前目錄更改到指定的項目文件夾下面,xxx替換為自己的工程文件目錄,并展示該目錄下的所有文件和目錄
import os
#改變當(dāng)前工作目錄到谷歌云盤的路徑
path="/content/drive/MyDrive/colab/xxx/"
os.chdir(path)
os.listdir(path)
3.2 硬件選擇
點擊代碼執(zhí)行程序,選擇更改運行時類型,將運行加速器改為GPU,也可以使用TPU運算,這里不詳敘述TPU及其代碼框架。
等到連接成功后,使用!nvidia-smi
命令可以查看當(dāng)前使用的GPU設(shè)備,樓主作為白嫖怪好幾次分到了Tesla T4(還行)
3.3 環(huán)境配置
Colab已經(jīng)默認(rèn)安裝了PyTorch的環(huán)境,但是有一些其他的環(huán)境包需要自己安裝。
注意:每次使用Colab前都需要配置環(huán)境,因為Colab在用戶離開后會清空用戶的文件和數(shù)據(jù)。
例如,使用pip安裝FFmpeg文章來源:http://www.zghlxwxcb.cn/news/detail-441052.html
!pip install ffmpeg
3.4 運行程序
使用python命令運行python程序,例如文章來源地址http://www.zghlxwxcb.cn/news/detail-441052.html
!python inference_video.py --exp=2 --video=50_2.mp4
到了這里,關(guān)于免費使用高性能的GPU和TPU—谷歌Colab使用教程的文章就介紹完了。如果您還想了解更多內(nèi)容,請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!