xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-13 07:47:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update README.md

Browse Source
This commit is contained in:
HGB 2024-11-25 17:32:00 +07:00 committed by GitHub
parent e62cb92da2
commit 37272872ca
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
README.md
View File

@ -1,7 +1,9 @@
#### TPU/XLA Devices
Users with TPU/XLA devices can install the PyTorch XLA stable build with the following command:
```pip install torch~=2.5.0 torch_xla[tpu]~=2.5.0 -f https://storage.googleapis.com/libtpu-releases/index.html```
```
pip install torch~=2.5.0 torch_xla[tpu]~=2.5.0 -f https://storage.googleapis.com/libtpu-releases/index.html
```
This is the command to install the nightly 2.6.0 which might have some performance improvements:
@ -13,4 +15,6 @@ pip install 'torch_xla[tpu] @ https://storage.googleapis.com/pytorch-xla-release
To get memory info for TPU devices, install the [tpu-info](https://github.com/AI-Hypercomputer/cloud-accelerator-diagnostics/tree/main/tpu_info) package with the following command:
```pip install tpu-info```
```
pip install tpu-info
```