pytorch
关注
分享
扫一扫
文章平均质量分 85
高精度计算机视觉
人工智能与视像技术，对技术有点执着
展开
-
Pytorch中检查cuda的信息
Check CUDA version in PyTorchgcptutorials.comPyTorchThis article explains how to check CUDA version, CUDA availability, number of available GPUs and other CUDA device related details in PyTorch.torch.cudapackage in PyTorch provides several methods t...转载 2022-03-10 11:01:50 · 1585 阅读 · 0 评论 -
安装OpenPCDet碰到的问题
Ubuntu18.04重新安装了，现在要把OpenPCDet也重新装上，按照官方的教程来，一路安装好了torch， numpy之类的东西，没碰到问题，之后 开始安装OpenPCDet，然后就碰到了下面的问题，问题1 IndexError: list index out of range(openmmlab) user@MS-7816:~/devAI/OpenPCDet$ python setup.py developNo CUDA runtime is found, using CUDA.原创 2021-12-02 11:15:56 · 8651 阅读 · 2 评论 -
EfficientNet Backbone结构解析 -- 以EfficientNet-B0为例说明
一般而言，不论我们是否要将该主干结构用于何种模型，一般都是在这个基础上进行的。例如，将在Yolact中用EfficientNet替换Resnet，可以在同等效果下让模型数据量降为原来的大约1/4到1/5左右。下面我列出一下主干结构。self._conv_stemConv2dStaticSamePadding( (conv): Conv2d(3, 32, kernel_size=(3, 3), stride=(2, 2), bias=False))self._blocks模原创 2021-10-12 21:04:28 · 3148 阅读 · 0 评论 -
Pytorch编译maskRCNN问题：cpp_extension.py:189: UserWarning: Error checking compiler version for cl...
先给出两个报错的内容：d:\Anaconda3\envs\torch\lib\site-packages\torch\utils\cpp_extension.py:189: UserWarning: Error checking compiler version for cl: [WinError 2] 系统找不到指定的文件。d:\Anaconda3\envs\torch\lib\site...原创 2019-09-14 16:17:59 · 38702 阅读 · 37 评论 -
Pytorch运行官方maskrcnn-benchmark报错: no instance of function template "THCCeilDiv" matches
Pytorch运行官方maskrcnn-benchmark， 问题和官网上的位置不同，我的报错在这个位置，SigmoidFocalLoss_cuda.cu，报错内容如下，....e:/AMaskRCNN/maskrcnn-benchmark/maskrcnn_benchmark/csrc/cuda/SigmoidFocalLoss_cuda.cu(120): error: no...原创 2019-02-25 18:10:05 · 3850 阅读 · 4 评论 -
模块visdom安装与使用
Facebook visdom的源就不说了，直接去github看一眼就行了。https://github.com/facebookresearch/visdom这里只记录最基本的安装与使用。visdom安装十分简单，可以直接pip进行安装pip install visdom根据说明，安装好以后，需要启动visdom服务python -m visdom.server或者...原创 2019-02-20 18:17:07 · 9272 阅读 · 2 评论 -
pytorch通过torch.utils.cpp_extension构建CUDA/C++拓展
注意这个和前面的《Python与C语言混合编程：通过distutils或setuptools实现的一个简单的C扩展》不同，这个是pytorch的扩展，不是python的扩展。在pytorch的utils中，集成了setuptools模块。官方文档在这里：https://pytorch.org/docs/master/cpp_extension.html中文说明在这里：https://p...转载 2019-02-20 11:26:49 · 28129 阅读 · 32 评论 -
Pytorch学习：手动计算一个back-propagation网络
如果想了解一下，手动计算的结果能否和pytorch计算的结果对应起来，那么下面的代码可能参考一下。这个例子来自官网，因为是示教的，网络结构极为简单https://pytorch.org/tutorials/beginner/examples_nn/two_layer_net_nn.html我对其中的部分源码进行了修改，加了一些手动计算的源码，并且每一个输出都写出了torch.nn网络计算结...原创 2019-02-11 16:25:50 · 1905 阅读 · 0 评论 -
Pytorch源码解析－－Ｃ扩展的一个简要说明
版本：Pytorh v1.0， python3.7要了解Pytorch的源码，最好先对CPython有个大致的了解，比如Python的多态是如何实现的。Pytorch的底层源码，很多都用c++实现，例如，torch._C就是一个非常典型的Ｃ＋＋模块，很多对象都会继承“_C”模块里的内容，如，class Tensor(torch._C._TensorBase): def __d...原创 2019-01-30 21:49:20 · 8546 阅读 · 1 评论 -
Pytorch报错(Windows平台)：from torch._C import * ImportError: DLL load failed: 找不到指定的模块。
这个错误十分奇怪，以前多次安装都没碰到过，原因可能和安装过程有关。貌似这段时间外网信号非常糟糕，我的安装过程有多次中断重启。如何安装 pytorch请参考：https://blog.youkuaiyun.com/tanmx219/article/details/82814738如果你的网络不好，经常需要安装本地已经下载下来的包，并且需要anaconda自动处理相关依赖，请重点参考其中的离线安装方法（2...原创 2019-02-15 17:13:35 · 36874 阅读 · 10 评论 -
pytorch源码解析：Tensor的生成过程（附：gdb对pytorch进行Ｃ语言级调试的过程）
pytorch v1.0准备工作首先我假设你已经安装好了pytorch的调试版本，如果没有，请参考：pytorch源码开发：在Ubuntu中的编译调试(C语言源码级调试)编译pytorch时一定要使用python setup.py build develop，还要设置DEBUG=1，不然无法进入源码。下面是一个gdb调试pytorch程序的过程，有兴趣的可以参考。我要说明的是，gdb...原创 2019-02-04 13:13:22 · 3381 阅读 · 0 评论 -
Pytorch 在 Ubuntu18.04 上的编译、安装、与问题
这里主要是develop开发版的编译与安装。如果不是开发版，安装是非常容易的，anaconda已经做得非常到位了，可参考该帖的后面部分,https://blog.youkuaiyun.com/tanmx219/article/details/82831964在自己编译的情况下，和官网相比，我每次重新编译总是缺些什么，所以决定记录一下，https://github.com/pytorch/pytorch...原创 2019-01-16 16:49:44 · 6609 阅读 · 0 评论 -
pytorch问题：ModuleNotFoundError: No module named 'torch'
问题今天碰到一个怪问题，明明各种包都已经安装好了，进入python也可以正常使用pytorch，但一进入ipython, jupyter notebook就无法使用pytorch,>>>import torch as t报错：ModuleNotFoundError: No module named 'torch'事发突然，不知何故，硬着头皮重新安装$ co...原创 2019-01-10 10:36:53 · 92783 阅读 · 16 评论 -
Pytorch问题：autograd与backward()及相关参数的理解
关于backward标量的问题问题提出在pytorch的官方教程中，有这么一段（我稍微修改了一些）import torch#x = torch.randn(3, requires_grad=True)x = torch.tensor([-1.0, -0.1, -0.05],requires_grad=True)print(x)y = x * 2index = 0while...原创 2019-01-09 12:48:46 · 11982 阅读 · 5 评论 -
Pytorch cifar10_tutorial.py问题BrokenPipeError: [Errno 32] Broken pipe
问题如果你用windows玩cifar10_tutorial.py,一定碰到过这个问题：BrokenPipeError: [Errno 32] Broken pipe When i :"run cifar10_tutorial.py先给解决方案，参考：https://github.com/pytorch/examples/issues/201分析用IPython运行，事个运行过程...原创 2019-01-09 08:56:56 · 5391 阅读 · 2 评论 -
pytorch源码开发：在Ubuntu中的编译调试(C语言源码级调试)
下载git clone --recursive https://github.com/pytorch/pytorchcd pytorch 非C语言级调试版安装如果不需要C语言高度，用anaconda安装，参考（https://github.com/pytorch/pytorch）过程如下，sudo apt-get install cmake lmdb-utilssudo...原创 2019-01-20 15:30:45 · 2799 阅读 · 2 评论 -
一些用于练习的几个较小的数据集
---------------------http://www.vision.caltech.edu/archive.htmlhttp://www.vision.caltech.edu/visipedia/CUB-200.htmlhttp://www.vision.caltech.edu/Image_Datasets/Caltech256/---------------------ht...原创 2018-09-29 16:52:56 · 2146 阅读 · 0 评论 -
机器学习备注：Windows + Anaconda3 + pytorch
Anaconda的安装先说一下，windows的运行机抽和linux完全不一样，由于无法使用sudo等命令，权限检查显得十分怪异。所以安装时最好不要安装在C盘，我一般都安装在D盘（如 D:/anaconda3）。如果你一定要装在C盘，那么个别软件包有的时候conda执行起命令来可能会非常缓慢，甚至无缘无故中断（源码没去研究过，所以也不明所以，原因只能连懵带猜，也有可能是因为一些你懂的原因，谁知...原创 2018-09-22 18:47:14 · 3082 阅读 · 0 评论