Package | 解决 Windows10 Anaconda环境 CUDA11.5 pytorch运行报错 No CUDA runtime is found

解决CUDA配置问题与PyTorch GPU环境搭建
本文详述了一位算法工程师在Windows环境下使用Anaconda配置CUDA和PyTorch遇到的问题及解决过程。通过排查发现是缺少cuDNN导致的错误,安装对应版本cuDNN并重新安装GPU版PyTorch后,成功启用CUDA。文章提供了确认CUDA环境、查找cuDNN路径和安装PyTorch的步骤,对遇到类似问题的读者有指导价值。

.

背景

在windows下使用anaconda管理多个环境(尤其用于模型训练),已成为算法工程师的一种基本技能。平时大部分时间都在用linux+docker,我也趁着这次的环境调试好好学习了一下anaconda的环境创建和管理。整体来说,anaconda的使用还是非常人性化的,具体官网请转至 all you need know about anaconda.

.

环境配置

如标题所示,实验的系统环境是windows 10,系统中已安装最新的CUDA v11.5。
conda环境为python 3.7,并使用以下指令进行pytorch安装:

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge

然后运行模型的训练代码,报错信息如下:

====================================================================
No CUDA runtime is found, using CUDA_HOME='your CUDA installation path\NVIDIA GPU Computing Toolkit\CUDA\v11.5’

Traceback (m

评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值