金乡网站建设哪家便宜,百度推广营销页,网站关键词免费优化,企业网站建设的一般要素A卡用户画图炼丹、跑大模型甚至是跑机器学习、深度学习的有福了~
注意#xff1a;ROCm目前仅限在Linux系统下可用#xff0c;Windows暂不支持 1. 安装ROCm
RX6000系列及以下显卡使用ROCm 5.4.2稳定版本命令
【支持包括桌面级AMD Radeon RX6950XT、RX6900XT、RX6800XT、RX6…A卡用户画图炼丹、跑大模型甚至是跑机器学习、深度学习的有福了~
注意ROCm目前仅限在Linux系统下可用Windows暂不支持 1. 安装ROCm
RX6000系列及以下显卡使用ROCm 5.4.2稳定版本命令
【支持包括桌面级AMD Radeon RX6950XT、RX6900XT、RX6800XT、RX6800、RX6750XT、RX6750GRE、RX6700XT、RX6700GRE、RX6700、RX6650XT、RX6600XT、RX6600、RX6500XT、RX6400等 移动端AMD Radeon RX6850M XT、RX6800M、RX6700M、RX6650M XT、RX6800S、RX6700S、RX6650M、RX6600S、RX6600M、RX6500M、RX6300M等】
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.4.2较新的RX7000系列显卡需要安装ROCm 5.6版本(加上--pre选项是查询最新的预发布或开发版本)
【支持包括桌面级AMD Radeon RX7900 XTX、RX7900 XT、RX7800 XT、RX7700 XT、Radeon RX7600等 移动端RX 7600M XT等】
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm5.62. 安装完成在python环境里验证cuda是否可用
import torch
torch.cuda.is_available()代码执行结果打印true说明是OK的
3. 运行你的项目之前有一项必要的环境变量配置
3.1 首先打开Linux配置文件
vim ~/.bashrc3.2 在末尾加上一行
若是RX6000系列及以下显卡复制粘贴这个 export HSA_OVERRIDE_GFX_VERSION10.3.0 若是RX7000系列显卡则复制粘贴这个 export HSA_OVERRIDE_GFX_VERSION11.0.0 3.3 保存配置并退出使配置立即生效
source ~/.bashrc全文完
