AMD GPU用户的福音。用AMD GPU学习人工智能吧。 pytorch 1.4 on ROCM 3.1 for ubuntu 18.04, ROCM 版本 3.1 预编译版本,直接pip install xxxx.whl 使用,省事省力。提前安装好rocm平台驱动。