AMD GPU驱动,ROCM,Pytorch安装教程(A卡6700xt)

这篇具有很好参考价值的文章主要介绍了AMD GPU驱动,ROCM,Pytorch安装教程(A卡6700xt)。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

我用的操作系统为ubuntu20.04,其他系统应该类似,只是命令稍有不同。

安装AMD GPU驱动

AMD驱动下载地址:https://www.amd.com/en/support/kb/release-notes/rn-amdgpu-unified-linux-22-20

进去之后找到你想安装的版本,这里是闭源驱动。

下载好后运行sudo apt install amdgpu-install_5.4.50403-1_all.deb

这个名字根据自己下载的文件进行更改,直接Tab就行。

如果报依赖错误可以尝试sudo apt-get -f install命令来修复依赖错误。

之后更新依赖sudo apt update.

安装ROCM

安装ROCM,这里要--no-dkms不安装内核,看资料好像现在的ubuntu自己带内核,所以加上好像会有问题。有知道的可以评论区讨论。

sudo amdgpu-install --usecase=rocm,hip,mllib --no-dkms

到这里驱动和ROCM就安装完了。

安装Pytorch

可以在pytorch官网找到安装命令

https://pytorch.org/

AMD GPU驱动,ROCM,Pytorch安装教程(A卡6700xt),Pytorch,pytorch,ubuntu,人工智能

 

例如我用pip安装

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm5.4.2

测试安装

打开python3

import torch

torch.cuda.is_available()

如果返回True

恭喜你,安装成功。

安装完之后可能还需要进行一些操作。

#添加环境
sudo usermod -a -G video $LOGNAME
sudo usermod -a -G render $LOGNAME

#在文件执行
vim ~/.bashrc #添加下面
export HSA_OVERRIDE_GFX_VERSION=10.3.0 #我这里的显卡是amd的RX6700xt
export LD_LIBRARY_PATH=/opt/rocm/lib
source ~/.bashrc#通重新执行刚修改的初始化文件
# 如果运行import之后报错找不到这些文件,复制一下链接文件
cd $LD_LIBRARY_PATH
cp librocblas.so.0 librocblas.so
cp libMIOpen.so.1 libMIOpen.so
cp librocrand.so.1 librocrand.so

其它问题:

Process finished with exit code 139 (interrupted by signal 11: SIGSEGV)
我还遇到了这个报错,原因是我没有添加
export HSA_OVERRIDE_GFX_VERSION=10.3.0 #我这里的显卡是amd的RX6700xt
export LD_LIBRARY_PATH=/opt/rocm/lib
到对应的用户下,只加了root,所以只有root能用。

有热心网友指出:

dkms可以安装。跟系统内核不冲突,只有在装多版本ROCm时不能装dkms。
export HSA_OVERRIDE_GFX_VERSION=10.3.0一般不用。我的同学时7900xtx被这个坑了一下,不用export就行。
也不用export LD_LIBRARY_PATH=/opt/rocm/lib ,hipconfig命令执行应该能获得准确的环境变量。rocmdev和mlsdk也可以安装一下。文章来源地址https://www.toymoban.com/news/detail-597714.html

到了这里,关于AMD GPU驱动,ROCM,Pytorch安装教程(A卡6700xt)的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • GPU版PyTorch对应安装教程

    ①算力:需要先知道你的显卡,之后根据官网表格进行对应,得到算力 ②CUDA driver version:电脑上显卡的硬件驱动 ③CUDA runtime version:pytorch官网上所显示的CUDA版本号 三者之间需要满足的关系: CUDA driver version ≥ CUDA runtime version (即:②≥③) CUDA runtime version得支持自己电脑G

    2024年02月14日
    浏览(33)
  • pytorch安装GPU版本 (Cuda12.1)教程

    使用本教程前,默认您已经安装并配置好了python3以上版本 Cuda下载地址 当前最高版本的Cuda是12.1 我安装的就是这个版本 小提示:自定义安装可以只选择安装Cuda Runtime。Nvidia全家桶不必全部安装。把全家桶全部安装完直接系统盘占了6G,很大的。 Pytorch下载地址 我之前使用的是

    2023年04月15日
    浏览(41)
  • GPU版本pytorch(Cuda12.1)安装教程

    我们通过Pytorch官网安装torch的时候,会发现常常由于网速问题安装不成功,下面提供一种简单的方法可以成功安装Cuda12.1,亲测有效。 目录 一、常规方法  二、有效方法 2.1 创建并激活虚拟环境 2.2 添加清华源 2.3 安装torch   通过Pytorch官网安装torch,网址链接如下:PyTorch  会

    2024年02月11日
    浏览(52)
  • pytorch安装GPU版本 (Cuda12.1)教程: Windows、Mac和Linux系统下GPU版PyTorch(CUDA 12.1)快速安装

    🌷🍁 博主 libin9iOak带您 Go to New World.✨🍁 🦄 个人主页——libin9iOak的博客🎐 🐳 《面试题大全》 文章图文并茂🦕生动形象🦖简单易学!欢迎大家来踩踩~🌺 🌊 《IDEA开发秘籍》学会IDEA常用操作,工作效率翻倍~💐 🪁🍁 希望本文能够给您带来一定的帮助🌸文章粗浅,敬

    2024年02月12日
    浏览(46)
  • 2023最新pytorch2.0 gpu cuda安装教程

    pytorch gpu是必须装的,因为cpu太慢了。 最新的pytorch版本是2.0 下面是安装教程。 首先,登录pytorch.org网站,看你要安装的cuda版本。 目前最新的支持cuda11.8,于是我选择了11.8 然后,登录cuda官网,下载cuda11.8 cuda最新版已经是12.1了。pytorch不支持!!!!!所以,找11.8版下载。

    2024年02月07日
    浏览(57)
  • 如何用conda安装PyTorch(windows、GPU)最全安装教程(cudatoolkit、python、PyTorch、Anaconda版本对应问题)(完美解决安装CPU而不是GPU的问题)

            安装PyTorch的开发环境:Anaconda+CUDA+cuDNN+PyCharm Community 1.1 版本选择 第一步就是最关键的版本对应问题(这决定你能否成功安装PyTorch,以及能否成功安装GPU版本的关键问题),可以这么说,版本不能对应好,后面有很大的问题,因此,我们要先确定版本的对应关系。(

    2024年02月07日
    浏览(52)
  • 全网最稳妥通用的GPU版本Pytorch安装教程(Anaconda虚拟环境)

    朋友们,你们有没有为安装GPU版本的pytorch而苦恼过?搜的明明是GPU的安装教程,一顿操作猛如虎,最后print(torch.cuda.is_available())结果居然是False。不要慌,接下来给你介绍一个在anaconda虚拟环境下安装的最稳妥的教程!!! CUDA 是一个运行平台,是用于“GPU通用计算” cudnn 深

    2024年02月07日
    浏览(228)
  • 在anaconda下安装pytorch + python3.8+GPU/CPU版本 详细教程

    没安装Anaconda的同学可以参考以下安装链接: https://blog.csdn.net/qq_45281807/article/details/112442577 按照安装CPU版本和GPU两个版本进行分类,一般运行程序建议使用CPU版本的,安装更方便。 注意!如果切换镜像后当出现下载不了的情况,就先切换默认源,然后再修改另一个可以使用的

    2024年01月19日
    浏览(71)
  • pytorch安装GPU版本 (Cuda12.1)教程: Windows、Mac和Linux系统快速安装指南

    🌷🍁 博主 libin9iOak带您 Go to New World.✨🍁 🦄 个人主页——libin9iOak的博客🎐 🐳 《面试题大全》 文章图文并茂🦕生动形象🦖简单易学!欢迎大家来踩踩~🌺 🌊 《IDEA开发秘籍》学会IDEA常用操作,工作效率翻倍~💐 🪁🍁 希望本文能够给您带来一定的帮助🌸文章粗浅,敬

    2024年02月16日
    浏览(92)
  • 集显安装pytorch教程,cuda和cudann环境配置torch-GPU版本看这一篇就够了

    集显看最下面 首先GPU安装教程 1.安装Anaconda (这个是可以安装环境和前置的软件),这个网址是清华源下载,可以去官网下载 Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror 2.安装CUDA (这个是显卡驱动,需要安装好以用显卡计算,缩短运行时间)(c

    2024年02月02日
    浏览(72)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包