一. 安装CUDA

二. 安装CUDNN

三,安装GPU版本pytorch

参考CUDA和CUDNN安装指导

          windows11系统下GPU版pytorch安装指导

根据网上的教程安装即可,需要注意:

1. 安装完CUDA和CUDNN后先查看环境变量是否自己生成,尽量安装路径不要变更(据网友说这个也不会占用太多空间)。

2. 安装CUDA和CUDNN尽量低于电脑GPU一个版本,免得出错(有网友说同版本会出错)。保持pytorch版本和CUDA保持一致即可。

Logo

昇腾计算产业是基于昇腾系列(HUAWEI Ascend)处理器和基础软件构建的全栈 AI计算基础设施、行业应用及服务,https://devpress.csdn.net/organization/setting/general/146749包括昇腾系列处理器、系列硬件、CANN、AI计算框架、应用使能、开发工具链、管理运维工具、行业应用及服务等全产业链

更多推荐