以不同顺序加载助焊剂和CuArray会导致错误

时间:2019-07-13 21:20:22

标签: julia cudnn

我正在尝试使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

有人对我可以尝试的方法有任何想法吗?

谢谢

1 个答案:

答案 0 :(得分:0)

因此,在拔掉头发后,我可以通过将export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/cuda/lib64添加到我的.bashrc文件中来使其工作。

我相信julia / Flux / CuArrays只是找到CUDNN Toolkit的好选择。