以不同的顺序加载 Flux 和 CuArrays 会导致错误

Question

我正在尝试让 Flux 和 CuArrays 在我使用 POP!_OS 的支持 GPU 的笔记本电脑上工作。我加载的第二个包出现错误。

我最初没有安装 CUDA 工具，但现在我安装了，但每个包都出现错误。

我的 cuda 信息是：

(base) ➜ ~ find / -type d -name cuda 2>/dev/null

/usr/lib/cuda-10.1/targets/x86_64-linux/include/thrust/system/cuda
/usr/local/cuda-10.1/targets/x86_64-linux/include/thrust/system/cuda
/home/kailukowiak/anaconda3/lib/python3.7/site-packages/numba/cuda
/home/kailukowiak/anaconda3/pkgs/numba-0.43.1-py37h962f231_0/lib/python3.7/site-packages/numba/cuda
/home/kailukowiak/.julia/packages/CUDAnative/gJDZI/src/device/cuda
/home/kailukowiak/.julia/packages/Flux/qXNjB/src/cuda
/home/kailukowiak/.julia/packages/Flux/qXNjB/test/cuda

如果我运行以下代码块在不同的 Julia 会话中

using CuArrays
using Flux

using Flux
using CuArrays

我得到：

┌ Warning: CUDNN is not installed, some functionality will not be available.
└ @ Flux.CUDA ~/.julia/packages/Flux/qXNjB/src/cuda/cuda.jl:35

但是，错误与不同的包裹有关，具体取决于订单。

Errors Depending on Order

有人知道我可以尝试什么吗？

谢谢

Answer 1

因此，经过相当多的努力之后，我能够通过将 export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/cuda/lib64 添加到我的 .bashrc 文件来使其正常工作。

我相信 julia/Flux/CuArrays 只是无法找到 CUDNN 工具包。

以不同的顺序加载 Flux 和 CuArrays 会导致错误

Loading Flux and CuArrays in Different Orders Causes Errors

julia

flux.jl