.
背景
在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
解决CUDA配置问题与PyTorch GPU环境搭建

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

被折叠的 条评论
为什么被折叠?



