本文主要是介绍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进行计算,说明安装成功
这篇关于ubuntu24下python3.9安装pytorch的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!