您现在的位置是：首页 >技术教程 >电脑配置PyTorch使用GPU方法网站首页技术教程

电脑配置PyTorch使用GPU方法

一只小小程序猿~ 2025-08-17 12:01:04
简介电脑配置PyTorch使用GPU方法

电脑配置PyTorch使用GPU方法

一、安装CUDA12.1(最新支持cudnn的版本)

1.先删除之前的CUDA
2.下载好cuda_12.1.1_windows_network.exe，一直下一步。
3.cmd输入

nvcc -V

判断是否安装成功
在这里插入图片描述

二、安装cuDNN 8.x 版本（CUDA12.1对应的版本）并设置环境变量

下载好并解压出来，将目录下的四个文件都复制到
C:Program FilesNVIDIA GPU Computing ToolkitCUDAv12.1
文件下（直接复制，有替换直接点替换）：

环境变量设置（四个目录）：

风语者！平时喜欢研究各种技术，目前在从事后端开发工作，热爱生活、热爱工作。

站长推荐