Pytorch

felix.shao2025-08-24

概述

安装 Pytorch gpu 版本

# pytorch 安装的 cuda 是独立的安装包，和其他的 cuda 没有关系
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

写个 python 测试 cuda 释放可用。

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

安装 Pytorch cpu 版本

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

安装通用库

pip install matplotlib