AMD | GPU | 深度学习 | 如何使用

embedded/2024/12/30 0:58:03/
  • 问题:我在复现代码的时候,发现自己只拥有AMD的GPU,对于一个硬件小白来说,怎么办呢?我想看看怎么使用;
  • 解决:
  1. 首先要安装支持AMD的GPU的pytorch,pytorch;
    在这里插入图片描述
  2. 使程序在安装了支持AMD的GPU的pytorch的python环境中运行代码,下述代码可以检测在现有的环境中是否可以调用AMD的GPU:
# AMD 
# use_roc = torch.cuda.is_available()  # 对于 AMD GPU,ROCm会通过CUDA接口暴露
# print(f"是否可以使用 ROCm:{use_roc}")
use_roc = torch.version.hip is not None and torch.cuda.is_available()
print(torch.version.hip)
print(f"是否可以使用 ROCm:{use_roc}")
print(f"PyTorch 版本:{torch.__version__}")
print(f"CUDA 可用性:{torch.cuda.is_available()}")
print(f"HIP 版本:{torch.version.hip}")

输出:
在这里插入图片描述

嘿嘿,之前一直失败,直到切换了正确的python环境才成功,原来一直都没有使用安装了支持rocm版本的pytorch!!!小迷糊😊


http://www.ppmy.cn/embedded/148518.html

相关文章

电力-恶意代码合集

ESS发送客户端信息 科东测试ESS外发的客户端上报数量不足,将数据库中已接收的客户端信息全部删除,需要ESS重新再此发送客户端的全部信息。 通过修改:vim /kingsoft/conf_gd_sync_info.lua 【里面记录了已外发的客户端信息,增量上报】 将内容修改为:{} 修改完成后重启ESS管…

Java:链接redis报错:NoSuchElementException: Unable to validate object

目录 前言报错信息排查1、确认redis密码设置是否有效2、确认程序配置文件,是否配置了正确的redis登录密码3、检测是否是redis持久化的问题4、确认程序读取到的redis密码没有乱码 原因解决 前言 一个已经上线的项目,生产环境的redis居然没有设置密码&…

最适合智能体的身份认证技术:对比OpenID Connect、API keys、did:wba

最适合智能体的身份认证技术:对比OpenID Connect、API keys、did:wba 智能体需要新的身份认证技术 智能体对身份认证技术提出了新的需求,其中最重要的一个就是互联互通,特别是让任意两个智能体都能够互联互通。 其中的原理很简单:…

创建vue2项目或vue3项目超详细!

一、安装Node (去官网下载所需版本即可,或私我发你安装包) 1、检查Node版本 二、创建vue2项目 1、安装vue脚手架 npm install -g vue/cli 或 yarn global add vue/cli 2、查看vue的版本 3、创建一个项目 3.1 进入桌面路径(创…

新闻网站的个性化推荐:机器学习的应用

3.1可行性分析 开发者在进行开发系统之前,都需要进行可行性分析,保证该系统能够被成功开发出来。 3.1.1技术可行性 开发该新闻网站所采用的技术是vue和MYSQL数据库。计算机专业的学生在学校期间已经比较系统的学习了很多编程方面的知识,同时也…

Blazor 直接读取并显示HTML 文件内容

如果你想在 Blazor 中直接读取 并显示HTML 文件,可以使用 .NET 的文件读取方法,比如 File.ReadAllText。Blazor Server(服务端)可以通过服务器的文件系统访问文件。 以下是针对 Blazor Server 的解决方案: 1. 使用 Fi…

docker 安装minio

docker pull minio/minio #启动 mkdir -p /root/minio/config mkdir -p /root/minio/datadocker run -d \--name minio \-p 9002:9000 \-p 9001:9001 \--restartalways \-v /root/minio/data:/data \-v /root/minio/config:/root/.minio \-e "MINIO_ACCESS_KEYminioadmin…

江苏计算机专转本 技能Mysql知识点总结(三)

四、视图与索引 视图是一张虚拟的表,不存在真正的数据,数据全部基于数据表,但是可以正常使用增删改查 创建视图 CREATE VIEW 视图名 AS SELECT column1, column2, ... FROM 表名 WHERE 表达式; 删除视图 DROP VIEW 视图名; 修改视…