- 博客(4)
- 收藏
- 关注
原创 【记录】Linux系统 pytorch使用其他显卡的方法
例如:指定id为‘2,4’显卡,那么后文代码中第一张显卡为‘cuda:0’,第二张显卡为‘cuda:1’;解决即使指定显卡,Pytorch仍在cuda:0上训练出现内存不够的问题。再如:指定id为‘3’的显卡,后文代码中的显卡号为‘cuda:0’.2、无论之前显卡的id是什么,要及时更新后续的代码段。引号里的数字代表显卡号。本篇主要详述用这个方法。
2023-10-10 17:08:18 166
原创 【记录】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 337
原创 【记录】将GPU中list类型的数据转移到CPU
将GPU变量转到cpu上。加torch.tensor(yourdata).detach().cpu().numpy(),可以解决这个问题。
2023-06-19 21:04:25 869
空空如也
transformers模型执行save_pretrained报错
2023-04-14
TA创建的收藏夹 TA关注的收藏夹
TA关注的人