在6650xt上跑pytorch

缘起:想 @跟李沐学AI , 但局限于自己只有一个amd6650xt显卡, 想尽量利用起来.很庆幸最后是找到了前辈们铺垫好的路, 再次感谢各位前辈的辛勤付出

引用:

[1] AMD显卡使用Pytorch深度学习配置(RX6600 ubuntu20.04为例)-2022.9.20  @饱腹的鼹鼠 https://zhuanlan.zhihu.com/p/566112395

[2] 03 安装【动手学深度学习v2】

[3] docker容器中使用jupyter notebook @理心炼丹 https://zhuanlan.zhihu.com/p/74243731

步骤:

1. 安装amd显卡驱动
   

2. 安装docker并拉取rocm/pytorch最新版镜像

3. 在物理机上下载d2l-zh.zip并解压

4. docker启动rocm/pytorch镜像, 设置目录映射,将物理机上d2l-zh的pytorch/ 文件夹设置为docker打开的工作目录; 设置端口映射 8888
   

5. 在镜像内安装jupyter(清华源), d2l(官方源); 给jupyter配置成可以远程连接
   

6. docker对rocm/pytorch镜像重新打包, 并

   1)暴露环境变量: HSA_OVERRIDE_GFX_VERSION=10.3.0

   2.启动运行: jupyter notebook --allow-root
   

7. 将docker启动命令写成shell文件,后续可以直接执行脚本启动jupyter

1.安装amd显卡驱动

参考[1] 即可

2.安装docker并拉取rocm/pytorch最新版镜像

参考[1] 即可

3.在物理机上下载d2l-zh.zip并解压

参考[2] 即可

4.docker启动rocm/pytorch镜像, 设置目录映射,将物理机上d2l-zh的pytorch/ 文件夹设置为docker打开的工作目录; 设置端口映射 8888

5.在镜像内安装jupyter(清华源), d2l(官方源); 给jupyter配置成可以远程连接

配置jupyter, 参考[3]

6.docker对rocm/pytorch镜像更新, 并使用Dockerfile再次构建

1)暴露环境变量: HSA_OVERRIDE_GFX_VERSION=10.3.0

2.启动运行: jupyter notebook --allow-root

第一次更新镜像

编辑Dockerfile

第二个构建镜像

7.将docker启动命令写成shell脚本文件,后续可以直接执行脚本启动jupyter

最后就是设置一下文件的可执行权限,然后就可以

sudo ./pytorch_start.sh

愉快的启动docker镜像啦