fix: torch

2024-09-12 13:06:24 -07:00
parent c52ca16334
commit 6c902d6f75
1 changed files with 2 additions and 0 deletions
@@ -59,6 +59,8 @@ sudo apt-get clean
 sudo rm -rf /var/lib/apt/lists/*
 EOF

+RUN python${PYTHON_VERSION} -m pip install /home/rocm/torch.whl
+
 # Torch Vision
 WORKDIR /home/rocm/
 RUN git clone --depth 1 https://github.com/pytorch/vision.git -b v0.17.2 vision