摘要: 如果一个机器上有多个gpu,可以使用多gpu训练。 一般数据量和模型比较大的时候训练速度会有明显的提升,模型和数据比较小的时候反而可能因为数据通信原因导致性能下降。 下面是一个简单的例子: import time import torch import torchvision.models from 阅读全文
posted @ 2024-03-31 17:53 Dsp Tian 阅读(11) 评论(0) 推荐(0) 编辑
摘要: 有时候pip下载安装特别慢,可以考虑更换下载源: 清华源: pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple 阿里源: pip config set global.index-url https://mi 阅读全文
posted @ 2024-03-31 15:46 Dsp Tian 阅读(7) 评论(0) 推荐(0) 编辑