wtongze/rocm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: torch

Browse Source
This commit is contained in:
wtongze
2024-09-12 13:06:24 -07:00
parent c52ca16334
commit 6c902d6f75
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch.dockerfile
+2
View File
@@ -59,6 +59,8 @@ sudo apt-get clean
sudo rm -rf /var/lib/apt/lists/* sudo rm -rf /var/lib/apt/lists/*
EOF EOF
RUN python${PYTHON_VERSION} -m pip install /home/rocm/torch.whl
# Torch Vision # Torch Vision
WORKDIR /home/rocm/ WORKDIR /home/rocm/
RUN git clone --depth 1 https://github.com/pytorch/vision.git -b v0.17.2 vision RUN git clone --depth 1 https://github.com/pytorch/vision.git -b v0.17.2 vision