草庐IT

mmcv与cuda,pytorch版本匹配要求

mmcv与cuda、pytorch版本兼容要求,见mmcv官方文档:https://mmcv.readthedocs.io/zh_CN/latest/get_started/installation.html#pip安装部分。目前网页上默认最新版2.x版本,若要切换旧版,点击页面左下角切换即可。查看自己的cuda和torch版本:python-c'importtorch;print(torch.__version__);print(torch.version.cuda)'#pytorch2.0版本需要cuda11.7及以上点击文档链接选择自己所需版本,拷贝对应的安装命令,进行安装注意!!!op

NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend解决办法

NotImplementedError:Couldnotrun'torchvision::nms'withargumentsfromthe'CUDA'backend.Thiscouldbebecausetheoperatordoesn'texistforthisbackend,orwasomittedduringtheselective/custombuildprocess(ifusingcustombuild).IfyouareaFacebookemployeeusingPyTorchonmobile,pleasevisithttps://fburl.com/ptmfixesforpossi

查看cuda cudnn 版本 & 一些常见疑问

一、快速确定版查看cuda   nvcc-V查看cudnn   dpkg-l|grepcudnn二、历史方法   有一些已经失效,这里仅仅作为备选查看cuda   方法一   nvcc-V或者nvcc—version   方法二   cat/usr/local/cuda/version.txt或者cat/usr/local/cuda/version.json查看cudnn   方法一   dpkg-l|grepcudnn   方法二   whereiscudnn_version或者whereiscudnn.h获得真实路径。这里以/usr/local/cuda/include/为例   cat/

C语言使用CUDA中cufft函数做GPU加速FFT运算,与调用fftw函数的FFT做运算速度对比

目录任务介绍环境所需相关软件下载与安装C语言:不调用库的GPU加速FFT代码C语言:调用fftw库的未使用GPU的FFT代码C语言:调用cufft库的GPU加速FFTgnuplot安装画图,maltab编写的FFT运算结果对比matlab测试信号和测试时的坑任务介绍时隔多年仍然逃不掉写C的命运……因为这个任务周期不短还踩了好多坑,必须记录一下了。任务简单要求就是使用C语言编写一个GPU加速的快速傅里叶变换(FFT)分为GPU加速的FFT代码改写、未使用GPU的FFT编写、运算速度对比、运算结果测试(与matlab结果对比),只要按照我文章写的顺序做就行环境所需相关软件下载与安装VisualSt

解决:RuntimeError: CUDA out of memory. Tried to allocate 160.00 MiB (GPU 0; 10.76 GiB total capacity..

完整报错:RuntimeError:CUDAoutofmemory.Triedtoallocate160.00MiB(GPU0;10.76GiBtotalcapacity;9.58GiBalreadyallocated;135.31MiBfree;9.61GiBreservedintotalbyPyTorch) 问题分析:内存分配不足:需要160MB,,但GPU只剩下135.31MB。解决办法:1.减小batch_size。注意batchsize的调整要配合学习率的调整,一般是正比关系,BS增大两倍,LR增大两倍或者根号二倍。减小也是相应更改。2.运行torch.cuda.empty_cach

CUDA编程第一章:windows下安装visual studio 2019+CUDA10.2的整体图文流程

目录简介下载链接及参考文章注意事项Visualstudio2019安装流程CUDA安装流程环境搭建结果验证自己进行CUDA编程验证总结简介去年虽然看了CUDA编程的基础知识(没学完つ﹏⊂),但是没有整理成笔记,并且一直没有使用,导致忘了好多。今年打算重新再把CUDA的基础知识学习一边,并进行总结梳理,记录成文,便于后续的复习。本篇博客是CUDA编程系列笔记的开篇,我打算先介绍下搭建CUDA编程环境的整体过程以及遇到的问题。我学习的参考书目为:樊哲勇老师的《CUDA编程基础与实践》。这本书我个人认为适合新手入门,书页不厚,内容通俗易懂,并且有C++代码示例。推荐一下。下载链接及参考文章1.vis

解决OSError: CUDA_HOME environment variable is not set. Please set it to your CU

解决OSError:CUDA_HOMEenvironmentvariableisnotset.PleasesetittoyourCU在windows。anaconda虚拟环境下安装pytorch的C++Extension的时候出现原因C++Extension有对CUDA的依赖,并且此cuda需要是电脑安装的而不是使用anaconda下载的cudatookit。具体原因请见:cuda和cudatoolkit_独孤的大山猫的博客-CSDN博客_cudatoolkit和cuda有关系吗在大多数情况下,上述cudatoolkit是可以满足Pytorch等框架的使用需求的。但对于一些特殊需求,如需要为P

CUDA编程模型系列三(矩阵乘)

CUDA编程模型系列三(矩阵乘)本系列教程将介绍具体的CUDA编程代码的细节CUDA编程模型系列三(矩阵乘)#include#include#defineBLOCK_SIZE32//errortype&event//a[][]*b[][]=c[][]////b00b01b02b03//b10b11b12b13//b20b21b22b23//b30b31b32b33////a00a01a02a03c00c01c02c03//a10a11a12a13c10c11c12c13//a20a21a22a23c20c21c22c23//a30a31a32a33c30c31c32c33////c21=a20

Win11基于WSL2安装CUDA、cuDNN和TensorRT(2023-03-01)

文章目录2023-03-06更新2023-03-05更新前言TensorRT介绍环境配置CUDADriver检查安装安装CUDA安装nvcc安装cuDNN安装验证安装TensorRT安装验证2023-03-06更新如果有小伙伴看了2023-03-05更新,发现设置环境变量后运行cuda代码在链接过程中仍然会有报错问题啥的,那我这里建议,先别管2023-03-05更新的内容了,还是按照我博客里的安装步骤一步一步往下安装,用sudoaptinstallnvidia-cuda-toolkit命令去安装nvcc,也不需要配置环境变量了。因为博客里的安装步骤是我亲自执行过的,在我的WSL2环境里是一点问

RuntimeError: CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling `cublasCreate(handle)`

背景:训练DialogueGPT(一个基于GPT2的生成模型)DialoGPT/data_loader.pyat457835e7d8acd08acf7f6f0e980f36fd327ea37c·microsoft/DialoGPT·GitHub遇到的问题:报错:RuntimeError:CUDAerror:CUBLAS_STATUS_NOT_INITIALIZEDwhencalling`cublasCreate(handle)`解决思路:我把输入用同样形状的随机张量进行了测试,发现用随机的整数张量可以,但是用我的输入就不行,于是想看看两者的区别到底是什么后来发现,DialogueGPT以及GP