PyTorch安装(不使用anaconda)

本文详细介绍了如何在Python中安装PyTorch的CPU和GPU版本,包括CPU版的直接安装方法,GPU版的CUDAToolkit下载和配置,以及版本检查和安装验证的过程。

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

一、概述

PyTorch的安装一般分为两种:CPU版、GPU版

区别在于GPU运算更快,需要额外下载CUDA

二、CPU版安装

https://pytorch.org/get-started/locally/

打开网址,找到对应的版本和对应的操作系统即可

需要注意自己的python版本

之后复制pip进行安装即可

安装完毕后导入torch模块进行验证

import torch
torch.cuda.is_available()
# 执行结果为False 代表是CPU版

三、GPU版

确保你的机器支持GPU版PyTorch

3.1 下载并安装CUDA Toolkit

CUDA(Compute Unified Device Architecture)是由 NVIDIA 提供的一个并行计算平台和编程模型,它允许开发者利用 NVIDIA GPU 的强大计算能力来加速计算密集型任务

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值