admin管理员组文章数量:1567745
环境
- 宿主机环境
ubuntu20.02(其实能跑docker的linux都可以)
CPU: AMD Ryzen 9 5950X
GPU: AMD 6700XT 12G理论上可以支持绝大部分的A卡 包含570 580
- 容器环境
ROCm5.0
pytorch1.10.0
python3.7
启动容器
docker run \
--device=/dev/kfd \
--device=/dev/dri \
--cap-add=SYS_PTRACE \
--security-opt seccomp=unconfined \
--ipc=host \
-dit \
--rm \
-e HSA_OVERRIDE_GFX_VERSION=10.3.0 \
amdih/pytorch:rocm5.0_ubuntu18.04_py3.7_pytorch_1.10.0
验证是否可用
docker exec -it 容器id bash
# 进入容器后输入一下命令
python
import torch
print(torch.cuda.is_available())
快速部署ssh直通版(ssh =直达=> 容器)
# 直接通过 exit run 的命令
# 并添加:2022端口ssh访问容器
# 设置root的密码为root
docker exec -it \
$(docker run \
--device=/dev/kfd \
--device=/dev/dri \
--cap-add=SYS_PTRACE \
--security-opt seccomp=unconfined \
--ipc=host \
-dit \
--rm \
-p 2022:22 \
-e HSA_OVERRIDE_GFX_VERSION=10.3.0 \
amdih/pytorch:rocm5.0_ubuntu18.04_py3.7_pytorch_1.10.0) \
bash -c \
"apt-get update && \
apt-get install openssh-server -y && \
echo 'PermitRootLogin yes' >> /etc/ssh/sshd_config && \
echo 'PasswordAuthentication yes' >> /etc/ssh/sshd_config && \
/etc/init.d/ssh start && \
echo 'root:root' | chpasswd"
验证1, docker exec验证
docker exec -it 容器ID python
import torch
print(torch.cuda.is_available())
验证2, ssh 验证
# 这里的2022是上面的 -p映射的
ssh root@容器宿主机IP -p2022
# 这里需要输入环境变量运行python
HSA_OVERRIDE_GFX_VERSION=10.3.0 python
import torch
print(torch.cuda.is_available())
版权声明:本文标题:pytorch + AMD卡 (docker快速构建A卡pytorch环境, 5分钟不折腾版) 内容由热心网友自发贡献,该文观点仅代表作者本人, 转载请联系作者并注明出处:https://www.elefans.com/dongtai/1726770509a1083814.html, 本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容,一经查实,本站将立刻删除。
发表评论