2023122始:
1. Bert模型的应用
环境搭建问题:笔记本自带是11.2的cuda, 所以选择11.1的pytorch,由于在pycharm上无法直接安装gpu版本的torch,所以选择创建虚拟conda环境安装,然后再去pycharm选择已存在的虚拟环境
conda install pytorch torchvision cudatoolkit=11.1 -c pytorch
tensorflow==3.4(比较稳定)
数据集合:lcqmc
2023122始:
1. Bert模型的应用
环境搭建问题:笔记本自带是11.2的cuda, 所以选择11.1的pytorch,由于在pycharm上无法直接安装gpu版本的torch,所以选择创建虚拟conda环境安装,然后再去pycharm选择已存在的虚拟环境
conda install pytorch torchvision cudatoolkit=11.1 -c pytorch
tensorflow==3.4(比较稳定)
数据集合:lcqmc