cuda
文章平均质量分 62
识久
这个作者很懒,什么都没留下…
展开
-
指定GPU训练
在终端训练代码签加上CUDA_VISIBLE_DEVICES=0即可。如下CUDA_VISIBLE_DEVICES=0 python train.py原创 2021-11-15 17:57:19 · 3389 阅读 · 0 评论 -
pytorch gpu版本的最全安装教程,含环境配置、CUDA(百度云下载)安装程序失败解决方案
看过我cpu版本的pytorch小白安装教程CPU版pytorch小白教程应该已经会设置清华源了,这里不再叙述,gpu版本的pytorch会默认以清华镜像源来下载,速度不用担心。请关闭杀毒软件 尤其360请关闭杀毒软件 尤其360请关闭杀毒软件 尤其360环境套餐:win10 + pytorch1.7.1 +vs2019+ CUDA10.1 + cuDNN8.0.4文章目录下载pytorch安装显卡驱动(已有驱动并且版本没问题的话可跳过此步)下载CUDA下载cuDNN安装CUDACUDA安装错误解原创 2021-01-10 11:43:54 · 38081 阅读 · 25 评论 -
pycharm跑pytorch时报错Process finished with exit code -1073741819 (0xC0000005)
环境配置python3.6+pytorch1.2.0+CUDA9.2+cuDNN7.1.4+cudatoolkit9.2+vs2019一切看上去都那么正常…跑程序的时候报了标题的错误,后悔没有截图纪念一下(当然也可以作死再换回去)调试工具走起,一步步定位到loss.backward()这一操作,就在这崩的,我心态也跟着崩了…找了好多方法,控制变量排除各种包的冲突numpy、matplotlib啊啥的,还有说相对路径改为绝对路径什么有的没的,但我数据集已经加载过了,明显不是这个问题。也要感谢nump原创 2021-01-10 01:39:42 · 4475 阅读 · 10 评论 -
如何在电脑上安装多个版本的CUDA
有时候有的项目要求很苛刻,有的需要CUDA>9.0,有的又要CUDA>10.0…无可奈何只能安装多个CUDA文章目录环境配置CUDA下载安装(已存在一个)cuDNN下载安装环境变量配置测试环境配置CUDA9.2,cuDNN7.1.4想了解如何配置正确的环境,避免包版本的冲突,可以到这里了解一下conda下载各种包时如何避免版本不匹配问题CUDA下载安装(已存在一个)gpu版pytorch安装教程,这里讲解了如何安装CUDA,如果我们想再装一个不同版本,继续看下去…CUDA下载直达链原创 2021-01-09 20:55:25 · 32563 阅读 · 15 评论