安装的时候看了很多博客,踩了一些雷,现在把成功下载的流程汇总下来。
假设这个时候已经安装好了ubuntu,我安装的是ubuntu22.04.
安装rocm
Linux® Drivers for AMD Radeon™ and Radeon PRO™ Graphics
可以点击上面这个链接,点击ubuntu x86 64-bit.
我选的是带rocm的版本
复制这里的代码,到终端运行。
sudo apt update
wget https://repo.radeon.com/amdgpu-install/6.2.3/ubuntu/jammy/amdgpu-install_6.2.60203-1_all.deb
sudo apt install ./amdgpu-install_6.2.60203-1_all.deb
sudo amdgpu-install -y --usecase=graphics,rocm
sudo usermod -a -G render,video $LOGNAME
默认是下载到/opt,这个文件有28g,请确保这个挂载下有足够的空间。
判断是否下载成功
可以打开文件,在/opt下搜索rocminfo和clinfo,能找到应该就安装成功了。下图是找到的rocminfo的结果。
添加环境变量
将刚刚找到的两个路径添加为环境变量,为这里两个文件是同一个路径。
echo 'export PATH=$PATH:/opt/rocm-6.2.3/bin | sudo tee -a /etc/profile.d/rocm.sh
弄好这个,就可以安装后续的软件了
安装anaconda
在anaconda官网点击Free Download,它会自动对应版本。
Anaconda | The Operating System for AI
可以参考这篇文章,写的很详细。
超详细Ubuntu安装Anaconda步骤+Anconda常用命令_ubuntu 安装anaconda-CSDN博客
下载完之后,可以进入图形化界面看看,在终端输入下面的命令。
anaconda-navigator
安装pytorch
激活环境
先查看python版本,然后创建环境并激活。
- 查看python版本
python --version
版本为python 3.12.7
2. 创建环境
conda create -n pytorch python=3.12.7
3.激活环境
conda activate pytorch
这样就进入了pytorch里。
4.下载pytorch
进入pytorch官网PyTorch往下拉,选择linux,pip,python,rocm 6.2.
在终端输入图中的命令
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.2
接下来就是静静地等待。
5.判断是否安装成功
import torch
print(torch.__version__)
print(torch.cuda.is_available())
print(torch.backends.cudnn.version())
看到True就说明成功了!
标签:python,Pytorch,AMD,rocm,pytorch,6.2,install,显卡,安装 From: https://blog.csdn.net/qq_59969626/article/details/143476638