cuda在ubuntu的安装使用分享

前言 之前给大家分享过opencv在jetson nano 2gb和ubuntu设备中使用并且展示了一些人脸识别等的小demo。但是对于图像处理,使用gpu加速是很常见 .(以下概念介绍内容来自百科和网络其他博主文章) GPU介绍(从GPU诞生之日起,GPU的设计逻辑与CPU的设计逻辑相差很多。GPU从诞生之日起,它的定位是3D图形渲染设备。在设计GPU时从其功能出发,把更多...

PyTorch 最新安装教程(2021-07-27)

PyTorch 最新安装教程(2021-07-27) 前言1. 安装 Anaconda2. 检查显卡,更新驱动3. 创建PyTorch环境4. 配置清华TUNA镜像源5. 安装 PyTorch6. 测试7. Pycharm使用Anaconda创建的pytorch虚拟环境后序 前言 万事开头难! 这句话又一次被我...

CUDA优化之PReLU性能调优

撰文|郑泽康 InsightFace模型里大量使用了PReLU激活函数,而PReLU的工作模式有两种: 1. PReLU(1),此时权重alpha的形状为(1, ),等价于一个Elementwise操作。 2. PReLU(channels),此时权重alpha的形状为(channe...

CUDA02 - 访存优化和Unified Memory

CUDA02 - 的内存调度与优化 前面一篇(传送门)简单介绍了CUDA的底层架构和一些线程调度方面的问题,但这只是整个CUDA的第一步,下一个问题在于数据的访存:包括数据以何种形式在CPU/GPU之间进行通信、迁移,以及在GPU内部进行存储、访问。 1 global 、shared 、constant、local 通常来讲,待计算的数据都存放在内存或者硬盘(外部存储设备)中,...

Ubuntu下安装CUDA

系列总目录 链接→ [系统配置、包/库安装、问题修复] 注意: Ubuntu18.04默认GCC-7,由于CUDA 9.x不支持GCC-7(下载页面没有对ubuntu18的支持),所以需要安装低版本的5或者<= 6.3.0,并设置为默认版本。否则应安装10.0以上版本CUDA ...

Python安装Pytorch教程(图文详解)

最近人工智能等多门课需要复现论文,近两年的论文很多都是Pytorch环境,所以,这里总结一下Pytorch的安装教程,做好最快、最简单、最好地完成安装。 本机环境Win10+1050Ti+Python3.7 1、查看本机的CUDA版本 cmd命令行输入nvidia-smi,在第一行最右边可...

cuda中threadIdx、blockIdx、blockDim和gridDim的使用

一、直观的感觉线程、线程块、线程格 为了直观的感觉线程、线程块、线程格,画了下面一个示意图。分为了两部分,一部分为线程格,另一部分为线程块,在图中线程格和线程块都画成了3维的,实际也可以是一维或者二维的。其中线程格里面最小的单元为线程块,而一个线程块里面最小的单元为线程。 二、thread...