Ubuntu、win10系统pytorch cuda版本安装

本文介绍了如何安装PyTorch深度学习框架,强调了在安装前需确保已配置好显卡驱动和CUDA。通过访问PyTorch官网选择相应版本并运行提供的命令进行安装。之后,通过运行验证代码确认PyTorch版本及GPU可用性。

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

深度学习常用pytorch框架,安装之前需要确保已经配置好显卡驱动和CUDA,具体教程后面再补充。

1.torch安装:
进入pytorch官网,选择对应版本进行安装,复制粘贴红框中的命令,进入prompt安装就可以了,效果如下:
在这里插入图片描述
2.安装验证:
验证安装版本和GPU是否可用:

import torch
print(torch.__version__)
print(torch.cuda.is_available())

out:
1.10.2
True
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值