问题说明:。我在Linux本地端运行pycuda的示例代码,没有任何问题。但是我在使用pycharm 远程调试代码时,出现了上述问题。
这个问题需要两步走,如果第一步之后,可以实现了,就不用第二步了
第一步:
export PATH="/usr/local/cuda/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda/lib64:$LD_LIBRARY_PATH"Steps
1. find .bashrc file.
2. Add above lines to it.
3. source .bashrc
4. To Test run command "nvcc --version"
有人这个地方使用的cuda-10.1(版本号),而我用的是cuda,因为我这里的cuda是cuda-10.1的软连接(相当于快捷方式)。如何所以“lrwxrwxrwx”中的第一个“l”表示软连接的意思。所以上述两种方式都可以。
第二步:
open compiler.py
add 下面的代码
nvcc = '/usr/local/cuda/bin/' + nvcc
如下所示:
def compile_plain(source, options, keep, nvcc, cache_dir, target="cubin"):
from os.path import join
assert target in ["cubin", "ptx", "fatbin"]
nvcc = '/usr/local/cuda/bin/' + nvcc # --> here is the new line
if cache_dir:
checksum = _new_md5()
...
compiler.py 的文件位置:
因为我有一个envs,所以我在其中一个envs下面找到的。你要可以用locate命令来定位。
anaconda3/envs/torch19/lib/python3.7/site-packages/pycuda
命令:
find ./lib/python3.7/site-packages -name compiler.py