pytools.prefork.ExecError: ошибка при вызове 'nvcc --version': [Errno 2] Нет такого файла или каталога - PullRequest
Новая
       109

pytools.prefork.ExecError: ошибка при вызове 'nvcc --version': [Errno 2] Нет такого файла или каталога

0 голосов
/ 19 мая 2019

У меня установлена ​​pycuda, и я пытаюсь проверить ее с помощью приведенного ниже кода. 

import pycuda.driver as cuda
import pycuda.autoinit
from pycuda.compiler import SourceModule
import numpy
a = numpy.random.randn(4,4)
a = a.astype(numpy.float32)
a_gpu = cuda.mem_alloc(a.nbytes)
cuda.memcpy_htod(a_gpu, a)

mod = SourceModule("""
  __global__ void doublify(float *a)
  {
    int idx = threadIdx.x + threadIdx.y*4;
    a[idx] *= 2;
  }
  """)

func = mod.get_function("doublify")
func(a_gpu, block=(4,4,1))

a_doubled = numpy.empty_like(a)
cuda.memcpy_dtoh(a_doubled, a_gpu)
print a_doubled
print a

Я получаю следующую ошибку:

pytools.prefork.ExecError: ошибка при вызове 'nvcc --version': [Errno 2] Нет такого файла или каталога

1 Ответ

0 голосов
/ 19 мая 2019

Работает после добавления строк в файл .bashrc

export PATH = / usr / local / cuda-10.1 / bin $ {PATH: +: $ {PATH}} $
экспорт LD_LIBRARY_PATH = / usr / local / cuda-10.1 / lib64 $ {LD_LIBRARY_PATH: +: $ {LD_LIBRARY_PATH}

Шаги
1. найти файл .bashrc.
2. Добавьте вышеуказанные строки к нему.
3. источник .bashrc
4. Проверить запуск команды "nvcc --version"

ссылка: https://askubuntu.com/questions/885610/nvcc-version-command-says-nvcc-is-not-installed помог

...