ubuntu24下python3.9安装pytorch

安装gpu版pytorch

1.去官网查找命令

https://pytorch.org/get-started/previous-versions/

由于我们安装的是CUDA12.1，所以我们选择12.1的进行安装

安装完成之后通过下面代码进行测试

>>> import torch
>>> print(torch.cuda.is_available())
True

显示True表明torch可以调用GPU进行计算，说明安装成功