torch==1.9.1+rocm4.0.1

AMD显卡安装pytorch

pytorch v1.8.0

ROCM 4.0.1 (Linux only)

pip install torch==1.8.0+rocm4.0.1 -f https://download.pytorch.org/whl/rocm4.0.1/torch_stable.html
pip install ninja
pip install 'git+https://github.com/pytorch/vision.git@v0.9.0'

pytorch v1.8.1

ROCM 4.0.1 (Linux only)

pip install torch1.8.1+rocm4.0.1 torchvision0.9.1+rocm4.0.1 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

ROCM 3.10 (Linux only)

pip install torch1.8.1+rocm3.10 torchvision0.9.1+rocm3.10 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

原文链接

https://pytorch.org/get-started/previous-versions/