gpt4 book ai didi

linux - Theano 搜索 CUDA 7.5 文件,但我安装了 CUDA 8。如何告诉它使用 CUDA 8?

转载 作者:太空宇宙 更新时间:2023-11-04 12:30:56 24 4
gpt4 key购买 nike

我使用 theano 0.9.0、CUDA 8.0.61、gcc 版本 4.9.3、NVIDIA UNIX x86_64 内核模块 375.39、Ubuntu 16.04.2 LTS,我有一个 geforce 1060。CUDA 的工作原理与我完成的测试相同。

但是每当我使用 theano ( http://deeplearning.net/software/theano/tutorial/using_gpu.html ) 进行 gpu 测试时,我都会收到此错误:ImportError: ('The following error happened while compiling the node', GpuElemwise{exp,no_inplace}((float64, vector)>) , '\n', 'libnvrtc.so.7.5: 无法打开共享对象文件:没有这样的文件或目录', '[GpuElemwise{exp,no_inplace}((float64, vector)>)]')

这真的很愚蠢,因为我有 CUDA 8,而不是 CUDA 7.5,所以它当然找不到 libnvrtc.so.7.5。我如何告诉 theano 我使用的是 CUDA 8 而不是 CUDA 7.5?

我的 .bashrc 设置过于复杂:

  • 导出 PATH="/usr/local/cuda-8.0/bin:$PATH"
  • 导出 LD_LIBRARY_PATH="/usr/local/cuda-8.0/lib64"
  • export CUDA_HOME=/usr/local/cuda-8.0
  • 导出 CUDA_ROOT=/usr/local/cuda-8.0
  • 导出 CPATH="$CPATH:~/.local/include"
  • 导出 CPATH="/usr/local/cuda-8.0/include:$CPATH"
  • 导出 LIBRARY_PATH="$LIBRARY_PATH:~/.local/lib"
  • 导出 LIBRARY_PATH="/usr/lib/nvidia-375:$LIBRARY_PATH"
  • 导出 LIBRARY_PATH="/usr/local/cuda-8.0/lib64:$LIBRARY_PATH"
  • 导出 LD_LIBRARY_PATH="$LD_LIBRARY_PATH:~/.local/lib"

我的 .theanorc 是:

[nvcc]

flags=-D_FORCE_INLINES

[全局]

设备=cuda0

[库达]

root =/usr/local/cuda-8.0/

最佳答案

你不能。安装 CUDA 7.5 或安装针对 CUDA 8 构建的框架版本。没有办法使您尝试做的事情起作用。

关于linux - Theano 搜索 CUDA 7.5 文件,但我安装了 CUDA 8。如何告诉它使用 CUDA 8?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/43308169/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com