Windows

windows安装nvidia显卡驱动、cuda、cudnn|PyTorch炼丹记

windowns环境中使用pytorch时候发现cuda不可用。记录全过程。 先前做过CentOS系统部署的系列文章,现在借此机会将windows系统的记录一下。 运行pytrch时候会报的错误有: AssertionError: Torch not compiled with CUDA enabled 或者 RuntimeError: No CUDA GPUs are available 解决方法 一、NVIDIA显卡驱动 1.查看本机NVIDIA显卡驱动 鼠标右键打开NVIDIA控制面板,系统信息-显示-驱动版本号 查看发现是388.16 有点老 2.下载nvidia显卡驱动 进入:nvidia官网 选择对应版本下载,安装 二、安装CUDA 1.下载和安装CUDA 进入:CUDA 因为pytorch最多支持到10.1,所以选择了10.1 在这儿我选择的是 CUDA Toolkit 10.1 update2 (Aug 2019) 下载(Select Target Platform → Windows → x86_64 → 10 → exe[network] → Base Installer → Download) 安装 ...