
简介
该用户还未填写简介
擅长的技术栈
未填写擅长的技术栈
可提供的服务
暂无可提供的服务
CUDNN error: CUDNN_STATUS_EXECUTION_FAILED解决方法
本人在ubuntu18.04系统复现执行cyclegan网络过程,使用显卡GeForce RTX 3090,该显卡仅仅支持使用cuda11以上的版本。由于之前配置官网指导环境pytorch1.4.0,误选用cuda10.1及cuda10.2,导致每次运行test.py过程极慢,半天过后报错CUDNN error: CUDNN_STATUS_EXECUTION_FAILED,后来,更换cuda11,
简单方法适用于pytorch的环境配置cuda及cudnn
然后直接一条命令配置好cuda11.3及cudnnconda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cudatoolkit=11.3 -c pytorch -c conda-forge或者这个命令:cuda11.1及对应的cudnnpip install torch==1.8.1+cu111 torchvision=
cityscapes数据集转化为coco数据格式
将cityscapes数据集转化为coco数据格式,主要是为了获取这几个文件,如图:此处,放上转好的文件,大家可以直接拿去用:百度网盘 链接: https://pan.baidu.com/s/1I1b5SbdNlWHUDV-mrEC6PQ 提取码: j826自己亲手制作的话,使用mmdetection中的tools里面包含的cityscapes.py文件,首先下载并按照mmdetection的要求

到底了







