- 博客(3)
- 资源 (7)
- 收藏
- 关注
RSS订阅转载 使用Pytorch在多GPU下保存和加载训练模型参数遇到的问题
原文转载至https://www.cnblogs.com/jinjunweina/p/12671833.html使用Pytorch在多GPU下保存和加载训练模型参数遇到的问题最近使用Pytorch在学习一个深度学习项目，在模型保存和加载过程中遇到了问题，最终通过在网卡查找资料得已解决，故以此记之，以备忘却。首先，是在使用多GPU进行模型训练的过程中，在保存模型参数时，应该使用类似如下代码进行保存：torch.save({ 'epoch': epoch,
2020-11-09 09:29:53
1850
原创 VNC VIEWER中遇到的问题
VNC VIEWER中遇到的问题1.启动vnc在服务器上输入vncserver或者vncserver -geometry 1366x768 :1# geometry设置vnc屏幕的大小，1为端口首先要求设置密码，password和verify，设置密码之后出现即启动成功，端口为1。输入ps -ef | grep -i vnc即可查询在使用vnc的进程。在电脑的vnc viewer上输入IP:端口，回车打开输入密码，即可连接到服务器桌面。2.unencrypted conne
2020-11-08 14:45:33
25405
6
原创 Linux服务器上安装anaconda和pytorch
Linux服务器上安装anaconda和pytorch一、前期准备1.查看linux的cuda版本在XShell中输入：nvcc -V即可查看cuda版本。但是注意，-V中的必须是大写，否则出错。2.查看服务器是否可以联网在XShell中输入：ping www.baidu.com如果出现...
2020-11-05 15:41:00
1405
1
database.zip
2020-06-09
flex和bison实现小型编译器生成中间代码源程序.zip
2019-07-03
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人