sdaujz的博客

机器上有八张显卡，占用了两三块，我选择了空余的GPU仍报错，使用下面代码查看可用GPU时，n_gpu为1，一脸问号n_gpu = torch.cuda.device_count()解决： 如下在主文件的头部加上这段代码，再次查看可用GPU时就为8了import osos.environ["CUDA_VISIBLE_DEVICES"] = "0,1,2,3,4,5,6,7"...

本地加载roberta-base模型文件，在该网站下载模型文件：roberta-base at main (huggingface.co)所需的有 config.json, merges.txt, pytorch_model.bin, vocab.json路径组织结构：└─model │ merges.txt │ vocab.json │ └─roberta-base config.json ...

Optimization :Optimization — transformers 3.5.0 documentation (huggingface.co)