Merge pull request pesser#1 from xeaon/xeaon-patch-1

update README.md to add addtitional steps for AMD cards

README.md

@@ -29,6 +29,14 @@ So run: <br>
 `conda env create -f environment.yaml` <br>
 `conda activate ldm`
 
+<h2 align="center">Additional steps for AMD Cards</h2>
+
+After activating your conda environment, you have to update torch and torchvision wheels which were built with ROCm support:
+
+`pip3 install --upgrade torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.1.1`
+
+<h2 align="center">Docker</h2>
+
 Alternatively, if you prefer to use Docker, you can do the following:
 
 1. Install [Docker](https://docs.docker.com/engine/install/)