AI說書 - 從0開始 - 135 | BERT 微調所需的相關依賴安裝

我想要一天分享一點「LLM從底層堆疊的技術」，並且每篇文章長度控制在三分鐘以內，讓大家不會壓力太大，但是又能夠每天成長一點。

現在來安裝 Transformer 模組：

try:
	import transformers
except:
	print("Installing transformers")
	!pip -q install transformers

接著匯入一些重要模組以供後續使用：

import torch
import torch.nn as nn
from torch.utils.data import TensorDataset, DataLoader, RandomSampler, SequentialSampler
from keras.utils import pad_sequences
from sklearn.model_selection import train_test_split
from transformers import BertTokenizer, BertConfig
from transformers import AdamW, BertForSequenceClassification, get_linear_schedule_with_warmup

from tqdm import tqdm, trange #for progress bars

import pandas as pd
import io
import numpy as np
import matplotlib.pyplot as plt
from IPython.display import Image #for image rendering

接著要來使用 GPU，偵測方式如下：

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
!nvidia-smi

結果如下：

注意上圖可能隨著 Google Colab 的更迭而有版本異動。