安装gpu版pytorch
1.去官网查找命令
bash
https://pytorch.org/get-started/previous-versions/
由于我们安装的是CUDA12.1,所以我们选择12.1的进行安装
安装完成之后通过下面代码进行测试
bash
>>> import torch
>>> print(torch.cuda.is_available())
True
显示True表明torch可以调用GPU进行计算,说明安装成功