mirror of
https://github.com/hanzoai/visor.git
synced 2026-09-24 17:04:14 +00:00
Lucas ManningandgVisor bot
2a3dc9b67b
The switch to a newer numpy was breaking guitar tests. PiperOrigin-RevId: 695514200
5 lines
205 B
Docker
5 lines
205 B
Docker
FROM python:3.10
|
|
|
|
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 |