Установка Pycuda на Google Colab - PullRequest
Новая
       33

Установка Pycuda на Google Colab

0 голосов
/ 26 июня 2018

Я хочу установить Pycuda на Google Colab. Я пытался pip install pycuda, но это дает WARNING: nvcc not in path. May need to set CUDA_INC_DIR for installation to succeed

Прочитав какой-то блог, я также сделал !export PATH=/usr/local/cuda/bin:$PATH и !export CUDA_ROOT=/usr/local/cuda, но все равно выдает ту же ошибку.

Кроме того, при завершении установки также отображается 

In file included from src/cpp/cuda.cpp:1:0:
src/cpp/cuda.hpp:14:10: fatal error: cuda.h: No such file or directory
#include <cuda.h>
          ^~~~~~~~
compilation terminated.
error: command 'x86_64-linux-gnu-gcc' failed with exit status 1

1 Ответ

0 голосов
/ 18 ноября 2018

Этот скрипт устанавливает пакеты, которые необходимы и для pycuda: 

!curl https://colab.chainer.org/install | sh -
...