win10+cuda+cuDNN+pytorch1.4.0
首先先说一下,由于cuda下载需要与本地的显卡驱动程序相匹配,所以先检查一下自己的驱动程序属什么版本,版本太低的话可以先更新一下(pytorch1.4目前只支持cuda9.0以上版本,所以显卡驱动程序版本太低也不行)

打开显卡驱动程序,系统信息中检查NvCUDA.DLL的版本。下面是不同版本所对应的cuda版本

显卡驱动程序更新方法如下:鼠标右键单击计算机-选择管理-在系统工具中-点击设备管理器-展开显示适配器-选中显卡驱动程序-右键单击它-选择更新驱动程序-弹出窗口-选择自动搜索更新选项-待安装完成即可。好接下来可以进行CUDA和pytorch_GPU的配置了。
1、先下载安装CUDA
下载地址:https://d

本文介绍了在Windows 10环境下,如何安装和配置CUDA 10.1、cuDNN,并详细讲解了PyTorch 1.4.0 GPU版本的安装步骤,包括检查NVIDIA驱动、下载安装CUDA、配置cuDNN,以及解决可能出现的问题,如torchvision的版本冲突。
最低0.47元/天 解锁文章
566

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



