踩坑
刃刃刀刀
这个作者很懒,什么都没留下…
展开
-
循环拼接tensor的简单操作###踩坑
循环拼接tensor1.创建空list2.将单个tensor转换为numpy,tensor_tmp3.将tensor_tmp压入listlist.append(tensor_tmp)4.转换list为tensortorch.Tensor(list)原创 2022-03-01 09:21:30 · 1494 阅读 · 0 评论 -
TypeError: can‘t convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to
Use Tensor.cpu() to copy the tensor to host memory first.Tensor在进行numpy计算的时候需要与numpy()的格式进行转换1.查看计算的tensor的位置信息。cpu tensor或者gpu tensor(cuda)2.如果为cuda tensor,需要进行转换numpy格式(cpu tensor原理一样)1)data.cpu().detach().numpy() #将data转换成cpu numpy格式2)result.cu原创 2021-04-09 12:45:42 · 1326 阅读 · 0 评论 -
RuntimeError: No CUDA GPUs are available
之前运行正常突然报错:RuntimeError: No CUDA GPUs are available// 报错信息RuntimeError: No CUDA GPUs are available解决方法// 检测一下cuda是否可用print(torch.cuda.is_available())False//重新在文件头加入下面说明import torchtorch.cuda.current_device()torch.cuda._initialized = True...原创 2021-03-16 09:53:57 · 8451 阅读 · 1 评论 -
2021-03-09踩坑 torch.cat
*将两张图像的detections tensor拟合在一起** 1. 通过循环检测后,两张图像的 tensor 存在 images_detections 中 **如: [tensor([[1,2], [2,3], [3,4]]), tensor([[1,2], [2,3], [3,4]])]** 2. 直接对 images_detections 执行自融合 torch.cat(images_detections, 0 ) 即可**如: tensor([[1,2], [2,3], [3,4], [1,原创 2021-03-09 21:25:14 · 352 阅读 · 2 评论 -
解决ubantu16.04 镜像源更换问题(vim操作)
解决ubantu16.04 镜像源更换问题(vim操作)**按照常见的解决办法更改 /etc/apt/sources.listvim命令常规操作 :su root:获取权限,vim /etc/apt/sources.list光标所在位置键入 :insert (改变操作模式{插入/替换})将网上提供的镜像源(如:清华源)粘贴。esc键入,退出前键入指令(:wq!)强制退出保存即可。正常使用指令: sudo apt-get update...原创 2020-11-18 22:14:04 · 354 阅读 · 2 评论 -
联想y7000p+ubuntu16.04安装显卡驱动后循环登录问题
联想y7000p+ubuntu16.04安装显卡驱动后循环登录问题每个人的问题不同,本人试过重装n次。最后尝试此方法有效。1.系统开机按F2 进入BIOS-> security 关闭secure boot 以及trust technology。2.进入boot ,Boot mode选UEFI 下面三个都是ENable3.保存,重启.本人认为是独显和集显的问题,至此显卡驱动正式安装。...原创 2020-11-17 10:26:47 · 341 阅读 · 0 评论