我正在尝试使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
但是,根据订单的不同,错误与不同的包装相关。
有人对我可以尝试的方法有任何想法吗?
谢谢
答案 0 :(得分:0)
因此,在拔掉头发后,我可以通过将export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/cuda/lib64
添加到我的.bashrc
文件中来使其工作。
我相信julia / Flux / CuArrays只是找到CUDNN Toolkit的好选择。