We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent dd42109 commit bbc1eb2Copy full SHA for bbc1eb2
1 file changed
docker/amd-docker.Dockerfile
@@ -35,7 +35,9 @@ RUN sudo apt update -y \
35
&& sudo apt update -y \
36
&& sudo apt install -y rocm-dev
37
38
-RUN pip install torch --index-url https://download.pytorch.org/whl/nightly/rocm6.3
+RUN pip install --upgrade pip
39
+
40
+RUN pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/nightly/rocm6.3
41
42
RUN pip install \
43
ninja \
0 commit comments