Pytorch
文章平均质量分 50
知识快到脑子里来
经验教训通通写进来!
展开
-
【记录-自用】Anoconda安装Pytorch-GPU踩坑点
Anoconda安装Pytorch-GPU踩坑点原创 2022-06-07 16:15:35 · 127 阅读 · 0 评论 -
【记录-自用】服务器跑代码
目录1 实验环境2 连接服务器2.1 MobaXterm安装2.2 连接操作3 服务器安装Anaconda4 PyCharm远程跑代码5 补充说明1 实验环境A、操作系统本地:Windows10 家庭版服务器:Linux x86B、工具1)PyCharm 2021.2.3(专业版)2)MobaXterm 用于服务器与本地电脑连接 下载地址2 连接服务器2.1 MobaXterm安装傻瓜式操作。安装路径修改一下2.2 连接操作打开软件,按照图片的箭头操作。下一个页面提示输入原创 2022-04-14 15:15:37 · 643 阅读 · 0 评论 -
python下载模块快速方法
**python下载模块快速方法!!!! **pip install 模块名 -i https://pypi.tuna.tsinghua.edu.cn/simple //清华源原创 2020-09-18 08:37:20 · 731 阅读 · 0 评论 -
基于GPU的Pytorch——CNN实现MNIST数据识别
CNN实现MNIST数据集分类时,发现CPU训练特别慢,20个epoch需要半个多小时,故修改为GPU训练,训练时长明显缩短。修改如下:需要把网络结构移到GPU上:model = Net()if torch.cuda.is_available(): print("The model is using GPU") model = model.cuda()训练数据集和测试数据集移到GPU上:for i, data in enumerate(train_loader):原创 2021-03-28 11:24:49 · 329 阅读 · 0 评论 -
记录贴——Pytorch出错问题合集
Pytorch框架的教训!!IndexError: invalid index of a 0-dim tensor. Use tensor.item() in Python or tensor.item<T>() in C++ to convert a 0-dim tensor to a number解决方法: 查找.data[0]全部替换成.item() (经常出现在loss函数)参考博客TypeError:Missing 1 required positional ar原创 2021-03-30 16:04:21 · 203 阅读 · 0 评论