摘要: #单个GPU进行训练 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") net.to(device) inputs, labels = inputs.to(device), labels.to(device 阅读全文
posted @ 2020-06-03 23:12 WuMinda 阅读(396) 评论(0) 推荐(0)
该文被密码保护。 阅读全文
posted @ 2020-06-03 14:41 WuMinda 阅读(2) 评论(0) 推荐(0)
摘要: #pytorch 历史版本 从此**链接**中可以查找pytorch历史版本的下载方法。如果感觉下载速度过慢,可以搭配镜像源使用。 pip install https://download.pytorch.org/whl/cu90/torch-1.0.0-cp36-cp36m-linux_x86_6 阅读全文
posted @ 2020-06-03 14:26 WuMinda 阅读(193) 评论(0) 推荐(0)