一、概述
PyTorch的安装一般分为两种:CPU版、GPU版
区别在于GPU运算更快,需要额外下载CUDA
二、CPU版安装
打开网址,找到对应的版本和对应的操作系统即可
需要注意自己的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 的强大计算能力来加速计算密集型任务