m0_47101748的博客

原创 cuda+cuDNN+pytorch_gpu安装

GPU版本安装需要显卡驱动，CUDA、cuDNN、python版本及pytorch各个版本需要对应兼容，并且在conda中新建一个虚拟环境安装pytorch。选择对应的11.5cuda版本，选择windows系统，64位，系统版本win11，选择本地安装类型，可下载到D盘，后续自定义安装至C盘。pytorch版本列表查看符合安装要求的版本，cuda11.5在官网中未查询到，pytorch v1.12.0支持cuda11.3版本，，cudnn可以理解未cuda的一个深度学习插件，放到制动文件夹中无需安装。