ubuntu24下python3.9安装pytorch

安装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进行计算,说明安装成功

相关推荐
千澜空13 分钟前
celery在django项目中实现并发任务和定时任务
python·django·celery·定时任务·异步任务
学习前端的小z16 分钟前
【AIGC】如何通过ChatGPT轻松制作个性化GPTs应用
人工智能·chatgpt·aigc
斯凯利.瑞恩21 分钟前
Python决策树、随机森林、朴素贝叶斯、KNN(K-最近邻居)分类分析银行拉新活动挖掘潜在贷款客户附数据代码
python·决策树·随机森林
yannan2019031342 分钟前
【算法】(Python)动态规划
python·算法·动态规划
埃菲尔铁塔_CV算法44 分钟前
人工智能图像算法:开启视觉新时代的钥匙
人工智能·算法
EasyCVR44 分钟前
EHOME视频平台EasyCVR视频融合平台使用OBS进行RTMP推流,WebRTC播放出现抖动、卡顿如何解决?
人工智能·算法·ffmpeg·音视频·webrtc·监控视频接入
打羽毛球吗️1 小时前
机器学习中的两种主要思路:数据驱动与模型驱动
人工智能·机器学习
蒙娜丽宁1 小时前
《Python OpenCV从菜鸟到高手》——零基础进阶,开启图像处理与计算机视觉的大门!
python·opencv·计算机视觉
光芒再现dev1 小时前
已解决,部署GPTSoVITS报错‘AsyncRequest‘ object has no attribute ‘_json_response_data‘
运维·python·gpt·语言模型·自然语言处理
好喜欢吃红柚子1 小时前
万字长文解读空间、通道注意力机制机制和超详细代码逐行分析(SE,CBAM,SGE,CA,ECA,TA)
人工智能·pytorch·python·计算机视觉·cnn