checking a potential fix

mlcommons · Mar 16, 2023 · 26080f7 · 26080f7
1 parent de2d404
commit 26080f7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile-ROCm b/Dockerfile-ROCm
@@ -7,7 +7,7 @@ LABEL version=1.0
 
 # The base image contains ROCm, python 3.8 and pytorch already, no need to install those
 RUN python3 -m pip install --upgrade pip
-RUN python3.8 -m pip install torch==1.13.1+rocm5.2 torchvision==0.14.1+rocm5.2 torchaudio --extra-index-url https://download.pytorch.org/whl/cu116
+RUN python3.8 -m pip install torch==1.13.1+rocm5.2 torchvision==0.14.1+rocm5.2 torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.2
 RUN python3 -m pip install --upgrade pip && python3 -m pip install openvino-dev==2022.1.0 opencv-python-headless mlcube_docker
 RUN apt-get update && apt-get install -y libgl1