AMD GPU驱动,ROCM,Pytorch安装教程(A卡6700xt) | 您所在的位置:网站首页 › 750ti的驱动版本 › 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/ Pytorch官网选择安装的版本例如我用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 $LOGNAMEsudo usermod -a -G render $LOGNAME #在文件执行vim ~/.bashrc #添加下面export HSA_OVERRIDE_GFX_VERSION=10.3.0 #我这里的显卡是amd的RX6700xtexport LD_LIBRARY_PATH=/opt/rocm/libsource ~/.bashrc#通重新执行刚修改的初始化文件# 如果运行import之后报错找不到这些文件,复制一下链接文件cd $LD_LIBRARY_PATHcp librocblas.so.0 librocblas.socp libMIOpen.so.1 libMIOpen.socp librocrand.so.1 librocrand.so 这里可以参考这个UP的视频。 【6700xt+rocm5.4安装tensorflow和pytorch】 |
CopyRight 2018-2019 实验室设备网 版权所有 |