支持AMD GPU —— 如何运行docker环境下的Jax环境

相关：
支持NVIDIA GPU —— 如何运行docker环境下的Jax环境



官方给出的安装主页：
https://hub.docker.com/r/rocm/jax


安装命令：

docker pull rocm/jax

image

image

posted on 2024-01-08 20:21  Angry_Panda  阅读(330)  评论(0)    收藏  举报

导航