人工智能
-
ubuntu tensorflow 和 pytorch 启动
1. 首先查看是否安装库，执行如下命令： 1 conda info –envs 2. 如果有，进行TensorFlow启动，执行如下命令： 1 source activate tf #这里的tf是1中命令执行完后的包的名称 3. 执行Python，在执行import，命令如下： 1 Python 2 import tf 效果如下： 4. py…
-
ubuntu下用anaconda快速安装 pytorch
1. 创建虚拟环境 1 conda create -n pytorch python=3.6 2. 激活虚拟环境 1 conda activate pytorch #这里 有用 source activate pytorch，因为我用的是conda激活的，这个看个人需求 3. 安装pytorch 打开pytorch官网https://pytorch.o…
-
[PyTorch] torch.squeee 和 torch.unsqueeze()
torch.squeeze torch.squeeze(input, dim=None, out=None) → Tensor 分为两种情况: 不指定维度 或 指定维度 不指定维度 input: (A, B, 1, C, 1, D) output: (A, B, C, D) Example >>> x = torch.zeros(2, 1,…
-
tesseract cuda pytorch安装 提升Tesseract-OCR输出的质量
tesseract下载地址：https://digi.bib.uni-mannheim.de/tesseract/ https://blog.csdn.net/u010454030/article/details/80515501 http://www.freeocr.net/ OpenCV OCR and text recognition wi…
-
pytorch转onnx常见问题
一、Type Error: Type ‘tensor(bool)’ of input parameter (121) of operator (ScatterND) in node (ScatterND_128) is invalid 问题模型转出成功后，用onnxruntime加载，出现不支持参数问题， 这里出现tensor(bool)是因为代码中使用了b…
-
pytorch自定义算子
参照官方教程，实现pytorch自定义算子。主要分为以下几步： 改写算子为torch C++版本 注册算子 编译算子生成库文件 调用自定义算子 一、改写算子 这里参照官网例子，结合openCV实现仿射变换，C++代码如下： 点击展开warpPerspective.cpp #include “torch/script.h” #include “opencv2/…
-
ubuntu20.04安装cuda10.2+pytorch+NVIDIA驱动安装+（Installation failed log: [ERROR]）
最近申请了服务器，需要自己去搭建环境，所以在此记录下自己的辛酸搭建历史，也为了以后自己不走弯路。话不多说直接搬运，因为我也是用的别人的方法，一路走下来很顺畅。 第一步首先安装英伟达驱动因为之前吃过亏，安装了ubuntu后直接装了cuda，结果没有任何效果，还连图形界面都出现不了（因为之前按照大佬们的攻略先一步禁用了ubuntu自带的显卡驱动，而自己又没有先装…
-
pytorch自定义网络层以及损失函数
转自：https://blog.csdn.net/dss_dssssd/article/details/82977170 https://blog.csdn.net/dss_dssssd/article/details/82980222 https://blog.csdn.net/dss_dssssd/article/details/84103834 …
-
pytorch中使用多显卡训练以及训练时报错：expect more than 1 value per channel when training, got input size..
pytorch在训练中使用多卡： conf.device = torch.device(‘cuda:0’ if torch.cuda.is_available() else “cpu”) conf.device_ids = list(conf.device_ids) self.model = torch.nn.DataParallel(self.model,…
-
pytorch保存模型等相关参数，利用torch.save()，以及读取保存之后的文件
转载自： https://www.cnblogs.com/qinduanyinghua/p/9311410.html 假设网络为model = Net(), optimizer = optim.Adam(model.parameters(), lr=args.lr), 假设在某个epoch，我们要保存模型参数，优化器参数以及epoch 一、 1. 先建立一个…
![ubuntu20.04安装cuda10.2+pytorch+NVIDIA驱动安装+（Installation failed log: [ERROR]）](https://pythonjishu.com/wp-content/uploads/2023/04/htcJPZCKoFlA20230408-480x300.jpg)
