1

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

但是，该错误与不同的包相关联，具体取决于订单。

错误取决于订单

有人对我可以尝试什么有任何想法吗？

谢谢

4

1 回答 1

1

所以在拉了很多头发之后，我能够通过添加export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/cuda/lib64到我的.bashrc文件来让它工作。

我相信 julia/Flux/CuArrays 很难找到 CUDNN 工具包。

于 2019-07-13T22:43:05.717 回答