search

pfnet / pytorch-pfn-extras

Supplementary components to accelerate research and development in PyTorch
https://medium.com/pytorch/migration-from-chainer-to-pytorch-8ed92c12c8
MIT License
268 stars 51 forks source link

Nightly CI failure #807

Closed kmaehashi closed 4 months ago

kmaehashi commented 4 months ago

Looks the problem is related to mpmath.

https://github.com/NVIDIA/TensorRT-LLM/pull/1146 https://pypi.org/project/mpmath/1.4.0a0/

(thanks @take-cheeze for investigation)

kmaehashi commented 4 months ago

https://github.com/pfnet/pytorch-pfn-extras/blob/8f03af2b6b4a81401b8519d21fe36123b24e7d40/.github/workflows/nightly-test-cpu.yml#L28

Let's replace --pre torch torchvision torchaudio with "torch>0.0.0a1" "torchvision>0.0.0a1" "torchaudio>0.0.0a1" to avoid using pre-releases for dependencies.