Win10下安装fastai2.2.7
目标:在笔记本上搭建自己的GPU环境,免费的GPU环境虽然好,但经常会出现断网的情况。
环境:Window10 Home 版本,GPU:NIVIDA MX350 2G,CUDA:11.0
初衷:fastai对开发人员来讲,真的是太友好了,找到一个合适的模型后(Bert,ResNet等),只需要经过装载数据、对模型进行微调,即可以完成一个当前最优秀模型的迁移学习,然后应该模型进行推理了。但是在安装的时候就卡了壳,该如何是好?
问题:直接pip install fastai 后
1.from fastai.text.all import * 报错
2.TextDataLoaders 报错
正确的姿势如下:
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
pip install fastai -i https://pypi.douban.com/simple
验证例子:
from fastai.text.all import *
path = untar_data(URLs.IMDB_SAMPLE)
df = pd.read_csv(path/'texts.csv')
dls = TextDataLoaders.from_df(df, path=path, text_col='text', label_col='label', valid_col='is_valid')
learn = text_classifier_learner(dls, AWD_LSTM)learn.predict("i like fastai very much")