pytorch
zhangxiangweide
趁年轻,越努力越幸运。
展开
-
训练yolov5出现的错误
yolov5和pytorch原创 2022-07-14 11:46:37 · 1946 阅读 · 3 评论 -
Kylin v10操作系统+ARM处理器+nvidia T4显卡编译GPU版本pytorch
ARM架构GPU 版本pytorch成功编译原创 2022-05-24 15:00:57 · 1359 阅读 · 0 评论 -
ARM架构上YOLOv5算法部署错误
报错:AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor‘ 的解决方案解决办法:降低PyTorch的版本到1.9.0No module named ‘skimage‘解决办法:pip install scikit-image原创 2022-05-16 20:44:15 · 543 阅读 · 0 评论 -
GPU compute capability(算力)和CUDA版本冲突问题
RuntimeError: CUDA error: no kernel image is available for execution on the deviceCUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.For debugging consider passing CUDA_LAUNCH_BLOCKING=1.原创 2022-01-13 14:32:02 · 6455 阅读 · 0 评论 -
英伟达显卡算力查询--A5000、RTX30系列以及GTX
显卡算力查询链接:https://developer.nvidia.com/zh-cn/cuda-gpus#computeCUDA GPUs | NVIDIA Developerhttps://developer.nvidia.com/cuda-gpus?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=18.04&target_type=deb_local#computep原创 2022-01-13 10:59:07 · 5002 阅读 · 0 评论 -
RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED
GTX 10系类显卡生成的容器CUDA10.2、cudnn7、Pytorch1.8.0、python=3.8可以训练。把镜像加载到RTX 20系类显卡容器训练出现上述错误:RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED目前解决办法:降低pytorch版本,PyTorch=1.7.1,python=3.7怀疑可能是:1.显卡驱动版本问题。2.显卡显存被占用太多。3.还可能是docker镜像并不能完美适应,10系类和20系类原创 2021-12-09 17:54:21 · 1205 阅读 · 0 评论 -
用pytorch训练EfficientNet网络。出现错误IndexError: dimension specified as 0 but tensor has no dimensions
if input.size(0) != target.size(0):IndexError: dimension specified as 0 but tensor has no dimensions原因:train或者validation dataset 样本数量与batchsize不是整除的关系解决:调整为整除关系,不再报错(从最大batchsize递减慢慢测试出来的)...原创 2021-07-23 09:36:04 · 451 阅读 · 0 评论 -
pytorch1.9.0 TypeError: __array__() takes 1 positional argument but 2 were given
#降低pillow的版本即可pip install pillow==8.2.0 TypeError: __array__() takes 1 positional argument but 2 were given#Steps to reproduce the behavior:import torchimport torchvision.transforms as transformsfrom PIL import Imagetfms = transforms.Compose(..原创 2021-07-21 09:53:07 · 508 阅读 · 0 评论 -
VScode配置libtorch环境(ubuntu18.04+libtorch1.7.0+pytorch1.7.0模型)
出现这个错误的原因是:terminate called after throwing an instance of 'c10::Error'。解答:下面三个.json里面libtorch和build的路径没有给对,是路径错误tasks.json{ "version": "2.0.0", "tasks": [ { "type": "shell", "label": "example-app", //lauch.json preLaunchTask "command": "原创 2021-05-19 14:47:36 · 1020 阅读 · 0 评论 -
把一个文件夹下的部分图像保存到另一个文件夹下
# -*- coding: utf-8 -*-import os.pathimport osimport cv2#将要转移的图像生成test.txt文本with open('H:/data/125penguan1/test.txt', 'r') as lines: for line in lines:#将每行中的多余空格删除 line = line.strip...原创 2018-11-18 17:15:19 · 1280 阅读 · 0 评论 -
如何安装指定版本Pytorch:
如何安装指定版本Pytorch:#使用conda安装指定版本conda install pytorch=0.1.10 -c soumith#使用pip安装指定版本pip install pytorch==0.1.10如何查看当前pytorch版本:import torchprint(torch.__version__)验证CUDA安装是否成功:#找到 /usr/local/...原创 2018-08-14 16:39:36 · 42536 阅读 · 2 评论