摘要:
#单个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)

浙公网安备 33010602011771号