pytorch
文章平均质量分 57
weixin_47343723
这个作者很懒,什么都没留下…
展开
-
指定GPU运行pytorch模型的n种方法
在Terminal终端输入(以下均以指定使用gpu1为例)或者直接。原创 2022-10-19 18:21:30 · 1227 阅读 · 0 评论 -
torch.cuda.is_available()返回False的问题
原因pytorch版本与cuda版本不匹配操作步骤1.查看cuda版本,cmd中输入nvcc -V2.网站*https://pytorch.org/get-started/previous-versions/*中查找该cuda版本对应的pytorch版本3.输入相应命令进行安装(1)新建环境conda create -n pytorch_gpu-cude9.0 python=3.7(2)进入环境activate pytorch_gpu-cude9.0(3) 粘贴对应Pytorch原创 2022-05-10 21:21:17 · 583 阅读 · 0 评论 -
pytorch基本函数练习
基础操作数据类型数据生成索引和切片维度变换进阶操作boardcast机制合并和分割(1)合并torch.cat() 函数,需保证除了合并维度外的其他维度size一致(2)分割torch.stack() 函数,保证两个合并张量size完全一致数学运算(1)运算符号运算+ 等价于 torch.add()加法- 等价于 torch.sub()减法* 等价于 torch.mul()乘法/ 等价于 torch.div()除法**乘方原创 2022-05-24 21:10:37 · 296 阅读 · 0 评论 -
记录torch.nn.CrossEntropyLoss()和torch.nn.BCELoss()区别
小白在这里弱弱记录一下区别,哪里不对请指教torch.nn.BCELoss()区别有以下三点:label是float类型输出为1维或者[batch,1]y_pred和label的大小保持一致x_data = torch.Tensor([[45,64],[66,88],[78,70],[55,23],[88,66],[44,34],[49,59],[64,62],[100,60]])y_data = torch.Tensor([[0],[1],[1],[0],[0],[1],[0]])原创 2022-05-12 18:39:08 · 793 阅读 · 4 评论