PyTorch安装(不使用anaconda)

一、概述

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 的强大计算能力来加速计算密集型任务。同时它提供了一系列的 API 和库，安装 CUDA Toolkit 确保了这些关键的库和 API 可用于 PyTorch，使其能够执行复杂的深度学习计算。

CUDA Toolkit 12.4 Downloads | NVIDIA Developer

 

1、选择对应的操作系统下载

2、下载完成后，不要修改安装目录，默认在C盘，修改安装目录可能会导致某些错误

3、安装时全部选择默认的选项

4、安装完成后重启即可

3.2 查看本机所支持最高的CUDA版本

nvidia-smi

3.3 下载对应的PyTorch

选择cuda版本时，不要超过本机所能支持的最高版本 

3.4 验证是否成功

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