- 博客(4)
- 收藏
- 关注
RSS订阅原创 【记录】Linux系统 pytorch使用其他显卡的方法
例如：指定id为‘2，4’显卡，那么后文代码中第一张显卡为‘cuda:0’，第二张显卡为‘cuda:1’；解决即使指定显卡，Pytorch仍在cuda:0上训练出现内存不够的问题。再如：指定id为‘3’的显卡，后文代码中的显卡号为‘cuda:0’.2、无论之前显卡的id是什么，要及时更新后续的代码段。引号里的数字代表显卡号。本篇主要详述用这个方法。
2023-10-10 17:08:18
143
原创 【记录】pip install git+https://github.com/XXX/XXX 报错
3、如果是 git+git:// 报错，将第二个git 换成http 或https。1、将 https 替换为 http。2、将 https 替换为 git。排列组合尝试，总有一个可行。
2023-10-09 21:27:17
293
原创 【记录】将GPU中list类型的数据转移到CPU
将GPU变量转到cpu上。加torch.tensor(yourdata).detach().cpu().numpy()，可以解决这个问题。
2023-06-19 21:04:25
816
空空如也
transformers模型执行save_pretrained报错
2023-04-14
TA创建的收藏夹 TA关注的收藏夹
TA关注的人