netbird/gvisor
0
0
Fork 0
mirror of https://github.com/netbirdio/gvisor.git synced 2026-05-22 17:12:49 -07:00
Code Issues Packages Projects Releases Wiki Activity

Update pytorch image python module versions.

Browse Source 
The switch to a newer numpy was breaking guitar tests.

PiperOrigin-RevId: 695514200
This commit is contained in:
Lucas Manning
2024-11-11 16:10:37 -08:00
committed by gVisor bot
parent 7920b5b40a
commit e22dc29727
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
images/tpu/pytorch/Dockerfile.x86_64
+1 -1
View File
@@ -1,5 +1,5 @@
FROM python:3.10
RUN pip install torch~=2.2.0 torch_xla[tpu]~=2.2.0 torchvision -f https://storage.googleapis.com/libtpu-releases/index.html
RUN pip install torch==2.5.1 torch_xla[tpu]==2.5.1 -f https://storage.googleapis.com/libtpu-releases/index.html
RUN git clone --depth=1 --branch v2.0.0 https://github.com/pytorch/xla.git