报错:
Traceback (most recent call last): File "setup.py", line 59, in <module> CUDA = locate_cuda() File "setup.py", line 56, in locate_cuda raise EnvironmentError('The CUDA %s path could not be located in %s' % (k, v)) EnvironmentError: The CUDA lib64 path could not be located in /usr/lib64 Makefile:2: recipe for target 'all' failed
gedit 打开 setup.py
cudaconfig = {'home':home, 'nvcc':nvcc, 'include': pjoin(home, 'include'), 'lib64': pjoin(home, 'lib64')}
把上面的最后一个lib64改为lib即可
cudaconfig = {'home':home, 'nvcc':nvcc, 'include': pjoin(home, 'include'), 'lib64': pjoin(home, 'lib')}