不要直接使用pycharm自带安装程序,如果你安装失败,往往是由于版本不兼容问题,如果可以尝试更新版本的话就更新版本,去查一下你当前环境适合哪个版本,如果你在使用了较多帖子仍然没有解决,建议自己写一个load_array( )函数,很简单
当然你也可以安装低版本的d2l包,但是里面可能会“”缺斤少两“”比如我下载的低版本就没有load_array这个函数,只能自己写一个用。
我一开始安装最新版本失败,但是旧版本可以,但是里面缺少部分包,你可以去搜搜其中方法,直接利用方法,相信我,这将比你去更改环境,调试环境快的多。
def load_array (data_arrays, batch_size, is_train=True):
"""构造一个PyTorch数据选代器"""
dataset = data.TensorDataset(data_arrays)
return data. DataLoader (dataset, batch_size, shuffle=is_train)
load_array( )是把dataset转换为tensor,然后返回迭代器的。其中data_arrays支持多个输入对象,到时候对应接收输出就可以了。
is_train决定了要不要打乱samples顺序。
你也可以使用Dataloader函数,但是没有这个函数好用