search

slimgroup / InvertibleNetworks.jl

A Julia framework for invertible neural networks
MIT License
149 stars 21 forks source link

GPU support of InvertibleNetworks in julia 1.6.1 #30

Closed ryxchina closed 3 years ago

ryxchina commented 3 years ago

Hi, I am trying to test the GPU support example for InvertibleNetworks in julia 1.6.1, but got errors. The error massages and package infomation is attached. Could someone help me with this? Thanks! Packages and error massages.txt

mloubout commented 3 years ago

This parameters error is a known issue with the recent CUDA.jl version and its dependencies. you can downgrade CUDA.jl for now (@alisiahkoohi you had this issue too right?)

alisiahkoohi commented 3 years ago

Yes I had. You could also explicitly install ExprTools@0.1.3 as suggested in https://discourse.julialang.org/t/help-needed-with-cuda/63923

ryxchina commented 3 years ago

Yes I had. You could also explicitly install ExprTools@0.1.3 as suggested in https://discourse.julialang.org/t/help-needed-with-cuda/63923

Thanks, @alisiahkoohi and @mloubout , it works like a charm!