PyTorch(GPU)安装与使用

PyTorch安装

之前的帖子：

Linux 安装 PyTorch - Excelius’s World

参考：

Pytorch(GPU版)安装 - 知乎 (zhihu.com)

1. 进入官网先择版本进行下载

PyTorch 官网（Start Locally | PyTorch），根据自己的情况选择合适的 PyTorch 版本：

这里我选择的是 Linux 系统下，CUDA 版本是 CUDA 12.4，最下方的便是下载的 pip 命令，但是建议使用国内镜像进行加速，清华大学镜像似乎没有 GPU，建议这里使用阿里云镜像进行加速：

pip3 install torch torchvision torchaudio -i https://mirrors.aliyun.com/pypi/simple/

2. 测试是否是 GPU 版本

在 PyCharm 中打开 Python 控制台：

然后输入以下命令：

import torch
torch.cuda.is_available()
torch.cuda.device_count()  

正常情况下，会和我差不多的情况，返回 True 和 GPU 的个数~