windows - 如何修改 .theanorc 以便 nvcc 在编译期间使用 -m64 标志？

Question

我按照 http://deeplearning.net/software/theano/install_windows.html#install-windows 中的步骤操作安装 theano 但遇到问题。其中之一是默认使用 http://deeplearning.net/software/theano/install_windows.html#install-windows 上的 .theanorc 设置。 ，在我的机器上 nvcc 尝试在 32 位模式下编译 theano，当我尝试在 python shell 上导入 theano 时看到以下内容(注意下面的 -m32):

['nvcc', '-shared', '-O3', '--use
/DEBUG', '-D HAVE_ROUND', '-m32',

然后它会遇到问题，因为它找不到 cublas.lib 在我的机器上确实不存在，在 32 位库下存储在 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v7.0\lib\Win32(此文件是否存在于其他用户系统的 32 位文件夹中？)。我有 cublas.lib 存储在 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v7.0\lib\x64 下，因此想在 64 位模式下编译.为此，我将 .theanorc 更改为:

[nvcc]
flags = -m64 --use-local-env  --cl-version=2013 

但这并没有达到预期的效果。 nvcc 仍在尝试以 32 位模式编译:

['nvcc', '-shared', '-O3', '--use-local-env', '--cl-version=2013', '-Xlinker', '
/DEBUG', '-D HAVE_ROUND', '-m32', '-Xcompiler', '-m64,-DCUDA_NDARRAY_CUH=a3b91bc

有谁知道如何修改 .theanorc 以便 nvcc 使用 -m64 flag 的正确语法？在编译期间？

--machine {32|64} -m Specify 32-bit vs. 64-bit architecture.

Allowed values for this option: 32, 64.

Answer 1

问题是您的 PYTHON 是 32 位 python。我们不支持混合使用 python、g++ 和 nvcc 位大小。这意味着它们都必须是 32 位或 64 位。

确保将它们全部安装为 64 位。

如果你想尝试支持这种混合情况，请检查 theano/sandbox/cuda/nvcc_compiler.py，我们在那里进行编译。这里我们添加 -m32 标志:

https://github.com/Theano/Theano/blob/master/theano/sandbox/cuda/nvcc_compiler.py#L324

如果你成功了，欢迎提交 PR 请求对 Theano 进行更改。