win10 安装pytorch的GPU版本全流程

最新推荐文章于 2024-11-11 17:22:42 发布

东洋 Dongyang

最新推荐文章于 2024-11-11 17:22:42 发布

阅读量260

点赞数

分类专栏：深度学习算法与原理文章标签： pytorch 深度学习 python

本文链接：https://blog.youkuaiyun.com/m0_46339652/article/details/119984049

版权

深度学习算法与原理专栏收录该内容

23 篇文章

订阅专栏

本文档详细介绍了CUDA、Cudnn的安装步骤，并提供了PyTorch环境配置及设备检测的测试代码，帮助读者成功搭建GPU加速的深度学习环境。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一. 安装CUDA

在这里插入图片描述

二. 安装Cudnn

在这里插入图片描述

三. 安装pytorch

在这里插入图片描述

测试

import torch
flag = torch.cuda.is_available()
print(flag)

ngpu= 1
# Decide which device we want to run on
device = torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")
print(device)
print(torch.cuda.get_device_name(0))
print(torch.rand(3,3).cuda())