How to install Pycuda on ubuntu

Pycuda lets you access Nvidia’s CUDA parallel computation API from Python

Installation

Use the package manager pip to install PYCUDA.

pip install pycuda

Alernate Method

Download latest pycuda

tar xfz pycuda-VERSION.tar.gz

cd pycuda-VERSION
python configure.py --cuda-root=/where/ever/you/installed/cuda

Here you can find cuda path on ubuntu /usr/local/cuda-10.0

make install

Testing the installation inside test directory using python/python3

python3 test_driver.py

If you still facing the issue. set environment variables manually

export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64

Error: Tensorflow cannot open libcuda-so-1

If you have the above error, Run your docker container with runtime as nvidia

docker run  `--runtime nvidia`  -it  Docker_image_name

I hope we were able to explain how to install pycuda on ubuntu and please feel free to contact support@clofus.com for any queries thanks