原因：是因为gpu显存太小而导致的错误。 解决方法： 1.将batch_size改小 2.不用gpu训练，改用cpu训练 import os os.environ["CUDA_VISIBLE_DEVICES"] = "-1" ...
分类：
其他好文 时间：
2021-03-16 13:56:03
阅读次数：
0
安装完gpu版本的tensorflow,导入正常，但是tf.test.is_gpu_available()一直返回False,解决办法： 1、打开NVIDIA控制面板，查看CUDA的驱动版本，如果版本太低，到Nvidia Experience更新驱动程序。默认安装即可。（安装过程自行百度） 2、如果 ...
分类：
其他好文 时间：
2021-03-10 13:07:41
阅读次数：
0
1.return torch.embedding(weight, input, padding_idx, scale_grad_by_freq, sparse)RuntimeError: Expected tensor for argument #1 'indices' to have scalar ...
分类：
其他好文 时间：
2021-03-01 13:14:12
阅读次数：
0
1 安装nvidia驱动 $ ubuntu-drivers devices == /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 == modalias : pci:v000010DEd00001C03sv00001B4Csd000011D7bc0 ...
分类：
系统相关 时间：
2021-02-22 12:22:33
阅读次数：
0
大家好，我是 zeroing~ 1，前言 之前谈到图片文本 OCR 识别时，写过一篇文章介绍了一个 Python 包 pytesseract ，具体内容可参考 介绍一个Python 包 ，几行代码可实现 OCR 文本识别！这篇文章 ，pytesseract 包是基于 Tesseract 封装得到的， ...
分类：
编程语言 时间：
2021-01-28 11:50:28
阅读次数：
0
#零、环境 Ubuntu 18.04 显卡 940MX 安装Cuda8.0+Cudnn5.1 由于Ubuntu 18.04自带的gcc版本为7.5.0,下文中使用了建立软链接的方式，使得Cuda使用gcc5.0。 #一、安装合适版本的显卡驱动 参考下图，选择适当版本 来源：https://docs. ...
分类：
系统相关 时间：
2021-01-28 11:39:10
阅读次数：
0
1 安装依赖 ## Install dependencies sudo apt -y install build-essential checkinstall cmake pkg-config yasm sudo apt -y install git gfortran sudo apt -y ins ...
分类：
系统相关 时间：
2021-01-13 10:46:58
阅读次数：
0
环境： pop_os(ubuntu18) cuda9 opencv3.2 GTX1070 darknet框架 官方网站：https://pjreddie.com/darknet/ 编译框架可能要根据自己的需要更改Makefile（我就根据我自己的情况更改了链接路径，因为我的opencv和cuda安装 ...
分类：
其他好文 时间：
2021-01-06 11:42:12
阅读次数：
0
1、查看自己的设备对应的cuda版本 （1）打开NVIDIA控制面板 （2）点击“帮助”，如下图： （3）点击“系统信息” >“组件”可以看到下图，CUDA版本是10.2： 2、在官网（https://pytorch.org/）上查看对应的版本 从下图可以看到红线标注的是需要运行的命令。但是，如果直 ...
适用于CUDA GPU的Numba 随机数生成 随机数生成 Numba提供了可以在GPU上执行的随机数生成算法。由于NVIDIA如何实现cuRAND的技术问题，Numba的GPU随机数生成器并非基于cuRAND。相反，Numba的GPU RNG是xoroshiro128 +算法的实现。xoroshi ...
分类：
其他好文 时间：
2020-12-31 12:02:59
阅读次数：
0
