Skip to content

Commit

Permalink
ROCm remove tensorflow
Browse files Browse the repository at this point in the history
GitHub Actions out of disk space
  • Loading branch information
YanWenKun committed May 27, 2024
1 parent 32f716b commit eb0b0b8
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion rocm/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,6 @@ RUN --mount=type=cache,target=/root/.cache/pip \
https://repo.radeon.com/rocm/manylinux/rocm-rel-6.1/torch-2.1.2%2Brocm6.1-cp310-cp310-linux_x86_64.whl \
https://repo.radeon.com/rocm/manylinux/rocm-rel-6.1/torchvision-0.16.1%2Brocm6.1-cp310-cp310-linux_x86_64.whl \
https://repo.radeon.com/rocm/manylinux/rocm-rel-6.1/pytorch_triton_rocm-2.1.0%2Brocm6.1.4d510c3a44-cp310-cp310-linux_x86_64.whl \
https://repo.radeon.com/rocm/manylinux/rocm-rel-6.1/tensorflow_rocm-2.15.0-cp310-cp310-manylinux2014_x86_64.whl \
--extra-index-url https://download.pytorch.org/whl/nightly/rocm6.1

# Dependencies for frequently-used
Expand Down

0 comments on commit eb0b0b8

Please sign in to comment.