摘要: 完美方案一: pytorch 用官方conda 安装成功后,就会自动的安装cuda和cudnn,唯一担心的就是网络被墙了 完美方案二: 在清华镜像源下载 pytorch 的tar.bz2的压缩包 conda install --offline torchxxxxx.tar.bz2 conda ins 阅读全文
posted @ 2020-06-29 20:15 yunshangyue 阅读(722) 评论(0) 推荐(0) 编辑
摘要: 1、最简单的方法 # 保存整个网络 torch.save(net, PATH) # 保存网络中的参数, 速度快,占空间少 torch.save(net.state_dict(),PATH) # #针对上面一般的保存方法,加载的方法分别是: model_dict=torch.load(PATH) mo 阅读全文
posted @ 2020-06-29 10:03 yunshangyue 阅读(125) 评论(0) 推荐(0) 编辑
摘要: cudnn加速运算 torch.backends.cudnn.enabled = True torch.backends.cudnn.benchmark = True 第一句话是说,使用的是非确定性算法,使用cudnn来寻找高效率的计算方式 第二句话是说,自动寻找最优的算法 一般情况下,当数据维度、 阅读全文
posted @ 2020-06-29 10:01 yunshangyue 阅读(388) 评论(0) 推荐(0) 编辑