search

deepjavalibrary / djl

An Engine-Agnostic Deep Learning Framework in Java
https://djl.ai
Apache License 2.0
4.13k stars 655 forks source link

0.18.0 on windows not working #1834

Closed joyol closed 2 years ago

joyol commented 2 years ago

on windows 7 cuda 102 0.17.0 is working 0.18.0 https://publish.djl.ai/pytorch/1.8.1/jnilib/0.18.0/win-x86_64/cpu/djl_torch.dll not found

frankfliu commented 2 years ago

@joyol

0.18.0 use pytorch 1.11.0 by default, on windows, pytorch 1.11.0 doesn't support cu102 any more, you have to use cuda 11.3 or 11.6, see: https://pytorch.org/get-started/locally/

You can use pytorch 1.10.0 with DJL 0.18.0 if you want to by setting environment variable:

set PYTORCH_VERSION=1.10.0
frankfliu commented 2 years ago

@joyol Looks like you are using pytorch 1.8.1, DJL 0.18.0 no longer support pytorch 1.8.1, you have to use 1.9.1, 1.10.0 or 1.11.0

joyol commented 2 years ago

thank you，windows7 maximum supported cuda 102