Skip to content

Commit bbc1eb2

Browse files
authored
upgrade pip and no cache
1 parent dd42109 commit bbc1eb2

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

docker/amd-docker.Dockerfile

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,9 @@ RUN sudo apt update -y \
3535
&& sudo apt update -y \
3636
&& sudo apt install -y rocm-dev
3737

38-
RUN pip install torch --index-url https://download.pytorch.org/whl/nightly/rocm6.3
38+
RUN pip install --upgrade pip
39+
40+
RUN pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/nightly/rocm6.3
3941

4042
RUN pip install \
4143
ninja \

0 commit comments

Comments
 (0)