使用GPU進行BERT微調-從0到1爬坑

原先的輕薄本bert微調一次要兩個小時,後面配置n卡臺式,希望用gpu來微調bert win10+anaconda+pycharm+tensorflow+GPU+bert 遇到很多坑,記錄下來以備後面使用。 之前對於gpu的使用是小白,本次從0開始,主要有以下幾個部分: 1、安裝anaconda 2、安裝cuda,cudnn 3、安裝tensorflow_gpu 4、使用gpu微調 這裏面涉及到了
相關文章
相關標籤/搜索