在amd显卡安装pytorch之前需要安装rocm 第一次失败 尝试https://gist.github.com/neolaw84/98876cebe0ad29d95050f11500c508fe 之后成功